diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-28 10:42:13 +0100 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-28 16:45:05 +0000 |
| commit | d109a4d0fb980f2e97c05cd86451eda0b53f37ef (patch) | |
| tree | 19092204016ccd60ea36637cc06a48a224508570 /include | |
| parent | 2f746b3f346d3efa9071fc53592652425869d6b3 (diff) | |
| download | armnn-d109a4d0fb980f2e97c05cd86451eda0b53f37ef.tar.gz | |
IVGCVSW-4932 Introduce ShapeInferenceMethod to TfLite Parser
* Introduced ShapeInferenceMethod::InferAndValidate to TfLiteParser
* Added unit tests
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Iad4aadce92912e7645b1652019ec4af478b7fc32
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnnTfLiteParser/ITfLiteParser.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/include/armnnTfLiteParser/ITfLiteParser.hpp b/include/armnnTfLiteParser/ITfLiteParser.hpp index de1eae7635..a68b719a66 100644 --- a/include/armnnTfLiteParser/ITfLiteParser.hpp +++ b/include/armnnTfLiteParser/ITfLiteParser.hpp @@ -28,9 +28,11 @@ public: struct TfLiteParserOptions { TfLiteParserOptions() - : m_StandInLayerForUnsupported(false) {} + : m_StandInLayerForUnsupported(false), + m_InferAndValidate(false) {} bool m_StandInLayerForUnsupported; + bool m_InferAndValidate; }; static ITfLiteParser* CreateRaw(const armnn::Optional<TfLiteParserOptions>& options = armnn::EmptyOptional()); |