aboutsummaryrefslogtreecommitdiff
path: root/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
diff options
context:
space:
mode:
authorDerek Lamberti <derek.lamberti@arm.com>2019-12-02 16:06:40 +0000
committerKevin May <kevin.may@arm.com>2019-12-03 11:27:56 +0000
commit05f1947ddbc3abf955acb7e52917dda859044bb6 (patch)
treeb864c7be7558a8d3f4f537289f339d642858eae5 /tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
parentdda061addba93a91eebb9d008615f4958b09e237 (diff)
downloadarmnn-05f1947ddbc3abf955acb7e52917dda859044bb6.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/NetworkExecutionUtils.hpp')
-rw-r--r--tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp7
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);
}