commit cdc495ea61a94ced93e877b62bcca5fa68f52f9b
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Sep 16 18:13:39 2021 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Sep 16 18:13:39 2021 +0100
tree 49b3dfeeb10afe08d5fda22d75c076ac17374f1c
parent f106ab745a12a5c773a9c315dcddef0c8bf11225

IVGCVSW-6382 Add Shape operator support to ONNX parser

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I3547effcbebf1ebc02d3b20f5db394a26991424d

src/armnnOnnxParser/OnnxParser.hpp

diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp
index f618ff4..101e99f 100644
--- a/src/armnnOnnxParser/OnnxParser.hpp
+++ b/src/armnnOnnxParser/OnnxParser.hpp
@@ -117,6 +117,7 @@
     void ParseFlatten(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 RegisterInputSlots(armnn::IConnectableLayer* layer, const std::vector<std::string>& tensorIndexes);