diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-09-16 18:13:39 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-09-16 18:13:39 +0100 |
| commit | cdc495ea61a94ced93e877b62bcca5fa68f52f9b (patch) | |
| tree | 49b3dfeeb10afe08d5fda22d75c076ac17374f1c /src/armnnOnnxParser/OnnxParser.hpp | |
| parent | f106ab745a12a5c773a9c315dcddef0c8bf11225 (diff) | |
| download | armnn-cdc495ea61a94ced93e877b62bcca5fa68f52f9b.tar.gz | |
IVGCVSW-6382 Add Shape operator support to ONNX parser
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I3547effcbebf1ebc02d3b20f5db394a26991424d
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 f618ff43fd..101e99ff8d 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -117,6 +117,7 @@ private: 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); |