Go to the documentation of this file.
24 #ifndef NNVM_GRAPH_ATTR_TYPES_H_
25 #define NNVM_GRAPH_ATTR_TYPES_H_
28 #include <unordered_map>
132 #endif // NNVM_GRAPH_ATTR_TYPES_H_
std::string JSONString
The result holder of JSON serializer.
Definition: graph_attr_types.h:46
std::vector< int > StorageVector
The result holder of storage id of each NodeEntry in the graph.
Definition: graph_attr_types.h:128
std::vector< TShape > ShapeVector
The result holder of shape of each NodeEntry in the graph.
Definition: graph_attr_types.h:61
Data structure Tuple and TShape to store dynamic sized shapes.
std::vector< int > DeviceVector
The result holder of device of each operator in the graph.
Definition: graph_attr_types.h:104
Layout expression. The layout is composed of upper cases, lower cases and numbers,...
std::vector< Layout > LayoutVector
The result holder of layout of each NodeEntry in the graph.
Definition: graph_attr_types.h:91
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::vector< int > DTypeVector
The result holder of type of each NodeEntry in the graph.
Definition: graph_attr_types.h:76