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/ExecuteNetwork | |
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/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 59269328c9..16e1fd3477 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -110,7 +110,9 @@ int main(int argc, const char* argv[]) ("file-only-external-profiling,g", po::bool_switch()->default_value(false), "If enabled then the 'file-only' test mode of external profiling will be enabled") ("counter-capture-period,u", po::value<uint32_t>(&counterCapturePeriod)->default_value(150u), - "If profiling is enabled in 'file-only' mode this is the capture period that will be used in the test"); + "If profiling is enabled in 'file-only' mode this is the capture period that will be used in the test") + ("parse-unsupported", po::bool_switch()->default_value(false), + "Add unsupported operators as stand-in layers (where supported by parser)"); } catch (const std::exception& e) { @@ -155,6 +157,8 @@ int main(int argc, const char* argv[]) bool printIntermediate = vm["print-intermediate-layers"].as<bool>(); bool enableExternalProfiling = vm["enable-external-profiling"].as<bool>(); bool fileOnlyExternalProfiling = vm["file-only-external-profiling"].as<bool>(); + bool parseUnsupported = vm["parse-unsupported"].as<bool>(); + // Check whether we have to load test cases from a file. if (CheckOption(vm, "test-cases")) @@ -202,7 +206,7 @@ int main(int argc, const char* argv[]) testCase.values.insert(testCase.values.begin(), executableName); results.push_back(std::async(std::launch::async, RunCsvTest, std::cref(testCase), std::cref(runtime), enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, - enableLayerDetails)); + enableLayerDetails, parseUnsupported)); } // Check results @@ -222,7 +226,7 @@ int main(int argc, const char* argv[]) testCase.values.insert(testCase.values.begin(), executableName); if (RunCsvTest(testCase, runtime, enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, - enableLayerDetails) != EXIT_SUCCESS) + enableLayerDetails, parseUnsupported) != EXIT_SUCCESS) { return EXIT_FAILURE; } @@ -268,6 +272,6 @@ int main(int argc, const char* argv[]) return RunTest(modelFormat, inputTensorShapes, computeDevices, dynamicBackendsPath, modelPath, inputNames, inputTensorDataFilePaths, inputTypes, quantizeInput, outputTypes, outputNames, outputTensorFiles, enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, - subgraphId, enableLayerDetails, runtime); + subgraphId, enableLayerDetails, parseUnsupported, runtime); } } |