From 05f1947ddbc3abf955acb7e52917dda859044bb6 Mon Sep 17 00:00:00 2001
From: Derek Lamberti <derek.lamberti@arm.com>
Date: Mon, 2 Dec 2019 16:06:40 +0000
Subject: IVGCVSW-4206 Optionally parse unsupported ops in ExecuteNetwork

Change-Id: I593e2540bd870d70aabb2c959f4e63a899967269
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
---
 tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp')

diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
index 2556a104b5..ff8b9af804 100644
--- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
+++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
@@ -371,6 +371,7 @@ struct ExecuteNetworkParams
     size_t                        m_SubgraphId;
     bool                          m_EnableLayerDetails = false;
     bool                          m_GenerateTensorData;
+    bool                          m_ParseUnsupported = false;
 };
 
 template<typename TParser, typename TDataType>
@@ -534,6 +535,7 @@ int RunTest(const std::string& format,
             bool printIntermediate,
             const size_t subgraphId,
             bool enableLayerDetails = false,
+            bool parseUnsupported = false,
             const std::shared_ptr<armnn::IRuntime>& runtime = nullptr)
 {
     std::string modelFormat = boost::trim_copy(format);
@@ -657,6 +659,7 @@ int RunTest(const std::string& format,
     params.m_SubgraphId               = subgraphId;
     params.m_EnableLayerDetails       = enableLayerDetails;
     params.m_GenerateTensorData       = inputTensorDataFilePathsVector.empty();
+    params.m_ParseUnsupported         = parseUnsupported;
 
     // Warn if ExecuteNetwork will generate dummy input data
     if (params.m_GenerateTensorData)
@@ -727,7 +730,7 @@ int RunTest(const std::string& format,
 
 int RunCsvTest(const armnnUtils::CsvRow &csvRow, const std::shared_ptr<armnn::IRuntime>& runtime,
                const bool enableProfiling, const bool enableFp16TurboMode, const double& thresholdTime,
-               const bool printIntermediate, bool enableLayerDetails = false)
+               const bool printIntermediate, bool enableLayerDetails = false, bool parseUnuspported = false)
 {
     std::string modelFormat;
     std::string modelPath;
@@ -841,5 +844,5 @@ int RunCsvTest(const armnnUtils::CsvRow &csvRow, const std::shared_ptr<armnn::IR
     return RunTest(modelFormat, inputTensorShapes, computeDevices, dynamicBackendsPath, modelPath, inputNames,
                    inputTensorDataFilePaths, inputTypes, quantizeInput, outputTypes, outputNames, outputTensorFiles,
                    enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, subgraphId,
-                   enableLayerDetails);
+                   enableLayerDetails, parseUnuspported);
 }
-- 
cgit v1.2.1