commit fe6aa2f1ec991bb42356ed4068158395e8c78a7c
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Sep 23 16:11:17 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Wed Sep 29 16:24:21 2021 +0000
tree 4246c8e850a1b66914eec0da5a7b4c613e07543d
parent cd20385b5511ffddb025066edc6988b824dfc1c4

IVGCVSW-6382 Add Unsqueeze operator support to ONNX parser

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

src/armnnOnnxParser/OnnxParser.hpp

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