diff options
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index b71b8dca49..196e903ed5 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -121,6 +121,7 @@ private: 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); |