Lines Matching full:onnx

11 #include <onnx/onnx.pb.h>
23 using ModelPtr = std::unique_ptr<onnx::ModelProto>;
28 using OperationParsingFunction = void(OnnxParserImpl::*)(const onnx::NodeProto& NodeProto);
32     using GraphPtr = std::unique_ptr<onnx::GraphProto>;
91     armnn::INetworkPtr CreateNetworkFromModel(onnx::ModelProto& model);
96     void SetupInfo(const google::protobuf::RepeatedPtrField<onnx::ValueInfoProto >* list);
102         const onnx::TensorProto::DataType& type = onnx::TensorProto::FLOAT);
107     void GetInputAndParam(const onnx::NodeProto& node,
119 …void AddConvLayerWithDepthwiseConv(const onnx::NodeProto& node, const armnn::Convolution2dDescript…
120 …void AddFullyConnected(const onnx::NodeProto& matmulNode, const onnx::NodeProto* addNode = nullptr…
121     void AddPoolingLayer(const onnx::NodeProto& nodeProto, armnn::Pooling2dDescriptor& desc);
129     void ParseActivation(const onnx::NodeProto& nodeProto, const armnn::ActivationFunction func);
130     void ParseClip(const onnx::NodeProto& nodeProto);
131     void ParseSigmoid(const onnx::NodeProto& nodeProto);
132     void ParseTanh(const onnx::NodeProto& nodeProto);
133     void ParseRelu(const onnx::NodeProto& nodeProto);
134     void ParseLeakyRelu(const onnx::NodeProto& nodeProto);
136     void ParseAdd(const onnx::NodeProto& nodeProto);
137     void ParseAveragePool(const onnx::NodeProto& nodeProto);
138     void ParseBatchNormalization(const onnx::NodeProto& node);
139     void ParseConcat(const onnx::NodeProto& nodeProto);
140     void ParseConstant(const onnx::NodeProto& nodeProto);
141     void ParseConv(const onnx::NodeProto& nodeProto);
142     void ParseFlatten(const onnx::NodeProto& node);
143     void ParseGather(const onnx::NodeProto& node);
144     void ParseGemm(const onnx::NodeProto& node);
145     void ParseGlobalAveragePool(const onnx::NodeProto& node);
146     void ParseMaxPool(const onnx::NodeProto& nodeProto);
147     void ParseShape(const onnx::NodeProto& node);
148     void ParseReshape(const onnx::NodeProto& nodeProto);
149     void ParseUnsqueeze(const onnx::NodeProto& nodeProto);
172     void ValidateInputs(const onnx::NodeProto& node,
186         std::unique_ptr<const onnx::TensorProto>    m_tensor;
187         onnx::TensorProto::DataType                 m_dtype;
189         OnnxTensor() : m_info(nullptr), m_tensor(nullptr), m_dtype(onnx::TensorProto::FLOAT) { }  in OnnxTensor()
195     /// map of onnx operation names to parsing member functions
212     std::unordered_map<std::string, std::pair<const onnx::NodeProto*, int>> m_OutputsMap;