From d109a4d0fb980f2e97c05cd86451eda0b53f37ef Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 28 Jul 2020 10:42:13 +0100 Subject: IVGCVSW-4932 Introduce ShapeInferenceMethod to TfLite Parser * Introduced ShapeInferenceMethod::InferAndValidate to TfLiteParser * Added unit tests Signed-off-by: Sadik Armagan Change-Id: Iad4aadce92912e7645b1652019ec4af478b7fc32 --- include/armnnTfLiteParser/ITfLiteParser.hpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'include/armnnTfLiteParser') 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& options = armnn::EmptyOptional()); -- cgit v1.2.1