25 #ifndef NNVM_GRAPH_ATTR_TYPES_H_ 26 #define NNVM_GRAPH_ATTR_TYPES_H_ 30 #include <unordered_map> 132 #endif // NNVM_GRAPH_ATTR_TYPES_H_
std::vector< int > StorageVector
The result holder of storage id of each NodeEntry in the graph.
Definition: graph_attr_types.h:128
std::vector< int > DeviceVector
The result holder of device of each operator in the graph.
Definition: graph_attr_types.h:104
std::unordered_map< std::string, int > DeviceAssignMap
The result holder of device of each operator in the graph.
Definition: graph_attr_types.h:112
std::string JSONString
The result holder of JSON serializer.
Definition: graph_attr_types.h:46
Data structure Tuple and TShape to store dynamic sized shapes.
std::vector< TShape > ShapeVector
The result holder of shape of each NodeEntry in the graph.
Definition: graph_attr_types.h:61
std::vector< int > DTypeVector
The result holder of type of each NodeEntry in the graph.
Definition: graph_attr_types.h:76
Layout expression. The layout is composed of upper cases, lower cases and numbers, where upper case indicates a (super-)dimension and the corresponding lower case with factor size indicates the split (sub-)dimension. For example, NCHW16c can describe a 5-D tensor of [batch_size, channel, height, width, channel_block]. Here sub-dimension channel_block=16 is the split of super-dimension C (channel).
std::vector< Layout > LayoutVector
The result holder of layout of each NodeEntry in the graph.
Definition: graph_attr_types.h:91