aboutsummaryrefslogtreecommitdiff
path: root/tests/InferenceModel.hpp
diff options
context:
space:
mode:
authorDerek Lamberti <derek.lamberti@arm.com>2019-12-02 16:06:40 +0000
committerDerek Lamberti <derek.lamberti@arm.com>2019-12-02 16:06:40 +0000
commit132563c7ec8a254c38882cd1d9debb6ded96cd70 (patch)
treeec51f851b1bdb10e5efcaf6df6f267b2dfcb0e01 /tests/InferenceModel.hpp
parentaa68e01f3eb0be2678ad8f4ef4e0a09ec7f43332 (diff)
downloadarmnn-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.hpp6
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 *){}};