Class NodeParser
Defined in File node_parser.h
Class Documentation
-
class NodeParser
NodeParser defined a base class for parsing node’s attributes.
Public Functions
-
NodeParser() = default
Constructor.
-
virtual ~NodeParser() = default
Destructor.
-
inline virtual ops::BaseOperatorPtr Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node)
Method to parse node of ONNX.
- Parameters
onnx_graph – [in] Define the onnx graph, which contains all information about the graph.
onnx_node – [in] Define the node to be resolved.
- Returns
PrimitiveC Attribute storage.
-
inline virtual ops::BaseOperatorPtr Parse(const caffe::LayerParameter &proto, const caffe::LayerParameter &weight)
Method to parse node of CAFFE.
- Parameters
proto – [in] Define the node which contains attributes.
weight – [in] Define the node which contains weight information.
- Returns
PrimitiveC Attribute storage.
-
inline virtual ops::BaseOperatorPtr Parse(const tensorflow::NodeDef &tf_op, const std::map<std::string, const tensorflow::NodeDef*> &tf_node_map, std::vector<std::string> *inputs, int *output_size)
Method to parse node of TF.
- Parameters
tf_op – [in] Define the node to be resolved.
tf_node_map – [in] Define the all nodes of the graph.
inputs – [in] Define the input name, that determines which inputs will be parsed including their order. Determined by user.
output_size – [in] Define the output num of current node, which need to be determined by user.
- Returns
PrimitiveC Attribute storage.
-
inline virtual ops::BaseOperatorPtr Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, const std::unique_ptr<tflite::SubGraphT> &tflite_subgraph, const std::unique_ptr<tflite::ModelT> &tflite_model)
Method to parse node of TFLITE.
- Parameters
tflite_op – [in] Define the node to be resolved.
tflite_subgraph – [in] Define the subgraph to be resolved.
tflite_model – [in] Define the model, which contains all information abort the graph.
- Returns
PrimitiveC Attribute storage.
-
NodeParser() = default