19 #include <onnx/onnx_pb.h>
22 #include <unordered_map>
24 #include "operator_set.hpp"
30 std::string get_node_domain(
const ONNX_NAMESPACE::NodeProto& node_proto);
36 explicit Model(
const ONNX_NAMESPACE::ModelProto& model_proto);
44 const std::string& get_producer_name()
const {
return m_model_proto->producer_name(); }
45 const ONNX_NAMESPACE::GraphProto& get_graph()
const {
return m_model_proto->graph(); }
46 std::int64_t get_model_version()
const {
return m_model_proto->model_version(); }
47 const std::string& get_producer_version()
const
49 return m_model_proto->producer_version();
78 const ONNX_NAMESPACE::ModelProto* m_model_proto;
79 std::unordered_map<std::string, OperatorSet> m_opset;
82 inline std::ostream& operator<<(std::ostream& outs,
const Model& model)
84 return (outs <<
"<Model: " << model.get_producer_name() <<
">");