mxnet
graph_attr_types.h
Go to the documentation of this file.
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  */
19 
24 #ifndef MXNET_GRAPH_ATTR_TYPES_H_
25 #define MXNET_GRAPH_ATTR_TYPES_H_
26 
27 #include <mxnet/op_attr_types.h>
28 #include <vector>
29 
30 namespace mxnet {
31 
45 using StorageTypeVector = std::vector<int>;
46 
60 using DispatchModeVector = std::vector<DispatchMode>;
61 
62 } // namespace mxnet
63 
64 #endif // MXNET_GRAPH_ATTR_TYPES_H_
std::vector< DispatchMode > DispatchModeVector
The result holder of dispatch mode of each Node in the graph.
Definition: graph_attr_types.h:60
namespace of mxnet
Definition: base.h:89
Additional operator attributes beside the ones provided by NNVM.
std::vector< int > StorageTypeVector
The result holder of storage type of each NodeEntry in the graph.
Definition: graph_attr_types.h:45