diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-12-02 16:06:40 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2019-12-02 16:06:40 +0000 |
commit | 132563c7ec8a254c38882cd1d9debb6ded96cd70 (patch) | |
tree | ec51f851b1bdb10e5efcaf6df6f267b2dfcb0e01 /tests/InferenceModel.hpp | |
parent | aa68e01f3eb0be2678ad8f4ef4e0a09ec7f43332 (diff) | |
download | armnn-132563c7ec8a254c38882cd1d9debb6ded96cd70.tar.gz |
IVGCVSW-4206 Optionally parse unsupported ops in ExecuteNetwork
Change-Id: I593e2540bd870d70aabb2c959f4e63a899967269
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 9e054c4375..6ec63ba223 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -93,6 +93,7 @@ struct Params bool m_VisualizePostOptimizationModel; bool m_EnableFp16TurboMode; bool m_PrintIntermediateLayers; + bool m_ParseUnsupported; Params() : m_ComputeDevices{} @@ -101,6 +102,7 @@ struct Params , m_VisualizePostOptimizationModel(false) , m_EnableFp16TurboMode(false) , m_PrintIntermediateLayers(false) + , m_ParseUnsupported(false) {} }; @@ -235,7 +237,9 @@ public: const std::string& modelPath = params.m_ModelPath; // Create a network from a file on disk - auto parser(IParser::Create()); + IParser::TfLiteParserOptions options; + options.m_StandInLayerForUnsupported = params.m_ParseUnsupported; + auto parser(IParser::Create(options)); armnn::INetworkPtr network{nullptr, [](armnn::INetwork *){}}; |