onnx.hpp

#pragma once
#include <dlprim/model.hpp>
namespace onnx {
    class TensorProto;
    class NodeProto;
}
namespace dlprim {
    
    class DLPRIM_API ONNXModel : public ModelBase {
    public:
        ONNXModel();
        virtual ~ONNXModel();
        void load(std::string const &file_name);
        virtual json::value const &network();
        virtual Tensor get_parameter(std::string const &name);
    private:
        void load_proto(std::string const &file_name);
        void prepare_network();
        void prepare_inputs_outputs();
        void parse_operators();
        void validate_outputs();
        void add_conv(onnx::NodeProto const &node);
        void add_ip(onnx::NodeProto const &node);
        void add_matmul(onnx::NodeProto const &node);
        void add_bn(onnx::NodeProto const &node);
        void add_operator(onnx::NodeProto const &node,json::value &v,bool add_outputs = true);
        void add_standard_activation(onnx::NodeProto const &node,std::string const &name,bool validate_inputs = true);
        void add_concat(onnx::NodeProto const &node);
        void add_softmax(onnx::NodeProto const &node);
        void add_elementwise(onnx::NodeProto const &node,std::string const &operation);
        void add_global_pooling(onnx::NodeProto const &node,std::string const &operation);
        void add_pool2d(onnx::NodeProto const &node,std::string const &operation);
        void add_flatten(onnx::NodeProto const &node);
        void add_clip(onnx::NodeProto const &node);
        void add_pad(onnx::NodeProto const &node);
        void add_bias(onnx::NodeProto const &node);
        void add_squeeze(onnx::NodeProto const &node);
        void add_reshape(onnx::NodeProto const &node);
        void handle_constant(onnx::NodeProto const &node);
        std::pair<std::string,Tensor> transpose_parameter(std::string const &name);


        template<typename T>
        T get_scalar_constant(std::string const &name);

        void check_outputs(onnx::NodeProto const &node,int minv,int maxv=-1);
        void check_inputs(onnx::NodeProto const &node,int inputs_min,int inputs_max=-1,int params_min=0,int params_max=-1);

        std::vector<int> tensor_to_intvec(Tensor t);
        void check_pad_op_2d(onnx::NodeProto const &node,std::vector<int> &pads);
        std::vector<int> get_pads_2d(onnx::NodeProto const &node);
        bool has_attr(onnx::NodeProto const &node,std::string const &name);
        template<typename T>
        T get_attr(onnx::NodeProto const &node,std::string const &name,T default_value);
        template<typename T>
        T get_attr(onnx::NodeProto const &node,std::string const &name);
        struct Data;
        std::unique_ptr<Data> d;
    };

}