diff options
| author | Sadik <sadik.armagan@arm.com> | 2018-09-19 15:30:00 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
| commit | b94967bae76196f28a86edc90b4bbe12618e2360 (patch) | |
| tree | 6cb277afad0348eb338c1ccdb5a0210d7062bd38 /src/armnnTfLiteParser/TfLiteParser.hpp | |
| parent | b9bf946704608c9a5e48189e338cc6efebae1d65 (diff) | |
| download | armnn-b94967bae76196f28a86edc90b4bbe12618e2360.tar.gz | |
IVGCVSW-1650 Add Support for Reshape layer on TF Lite parser
* Added Reshape operator support for the TfLite Parser.
Change-Id: I64a5650dac089905a402be4a9cb6032aa0d81f00
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
| -rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index b9f81e4118..f949484a4f 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -74,7 +74,8 @@ public: static BufferRawPtr GetBuffer(const ModelPtr& model, size_t bufferIndex); static armnn::TensorInfo OutputShapeOfSqueeze(const std::vector<uint32_t> & squeezeDims, const armnn::TensorInfo & inputTensorInfo); - + static armnn::TensorInfo OutputShapeOfReshape(const armnn::TensorInfo & inputTensorInfo, + const std::vector<int32_t> & targetDimsIn); private: // No copying allowed until it is wanted and properly implemented @@ -95,6 +96,7 @@ private: void ParseSqueeze(size_t subgraphIndex, size_t operatorIndex); void ParseRelu(size_t subgraphIndex, size_t operatorIndex); void ParseRelu6(size_t subgraphIndex, size_t operatorIndex); + void ParseReshape(size_t subgraphIndex, size_t operatorIndex); void RegisterProducerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IOutputSlot* slot); void RegisterConsumerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IInputSlot* slot); |