diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-09-30 12:10:50 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-10-08 16:53:53 +0000 |
| commit | 1112b016e7ffad979b7bd0c8d54c9c679d4043e2 (patch) | |
| tree | 9a951835f6f4dc0cd6b05517696ea69c25a03e3d /src/armnnOnnxParser/OnnxParser.hpp | |
| parent | 8636bc705cc33fd869f64ebf24b14836d5a40b29 (diff) | |
| download | armnn-1112b016e7ffad979b7bd0c8d54c9c679d4043e2.tar.gz | |
IVGCVSW-6449 Add GEMM operator support to ONNX parser
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I3c6979c72d44a15fb2dc3afc22ac30d1428684b0
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
| -rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index d388f501d4..ec19006be7 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -120,12 +120,16 @@ private: void ParseConv(const onnx::NodeProto& nodeProto); void ParseFlatten(const onnx::NodeProto& node); void ParseGather(const onnx::NodeProto& node); + void ParseGemm(const onnx::NodeProto& node); void ParseGlobalAveragePool(const onnx::NodeProto& node); void ParseMaxPool(const onnx::NodeProto& nodeProto); void ParseShape(const onnx::NodeProto& node); void ParseReshape(const onnx::NodeProto& nodeProto); void ParseUnsqueeze(const onnx::NodeProto& nodeProto); + void RegisterInputSlot(armnn::IConnectableLayer* layer, + const std::string& tensorId, + unsigned int slotIndex); void RegisterInputSlots(armnn::IConnectableLayer* layer, const std::vector<std::string>& tensorIndexes); void RegisterOutputSlots(armnn::IConnectableLayer* layer, const std::vector<std::string>& tensorIndexes); |