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/NetworkExecutionUtils | |
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/NetworkExecutionUtils')
-rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
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); } |