diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-14 10:02:22 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-29 13:05:35 +0000 |
commit | a9c2ce123a6a5a68728d040a0323c482bbe46903 (patch) | |
tree | 18f0556645c4228ae3551c9dd7a62a3b7154e93b /tests/InferenceModel.hpp | |
parent | 57512af8c7a628097c644fdae276cf0f6da336ee (diff) | |
download | armnn-a9c2ce123a6a5a68728d040a0323c482bbe46903.tar.gz |
IVGCVSW-4980 Introduce InferAndValidate option to ExecuteNetwork for parsers
* Introduced infer-output-shape option to TfLiteParser in ExecuteNetwork app
!armnn:3591
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I30bd5e51ac2b6759169e22a44586fd97986f2402
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 5588d55aaf..68ee8ae81a 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -95,6 +95,7 @@ struct Params bool m_EnableBf16TurboMode; bool m_PrintIntermediateLayers; bool m_ParseUnsupported; + bool m_InferOutputShape; Params() : m_ComputeDevices{} @@ -105,6 +106,7 @@ struct Params , m_EnableBf16TurboMode(false) , m_PrintIntermediateLayers(false) , m_ParseUnsupported(false) + , m_InferOutputShape(false) {} }; @@ -241,6 +243,7 @@ public: // Create a network from a file on disk IParser::TfLiteParserOptions options; options.m_StandInLayerForUnsupported = params.m_ParseUnsupported; + options.m_InferAndValidate = params.m_InferOutputShape; auto parser(IParser::Create(options)); armnn::INetworkPtr network{nullptr, [](armnn::INetwork *){}}; |