diff options
Diffstat (limited to 'tests/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 0761551762..828d24985b 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -86,6 +86,8 @@ int main(int argc, const char* argv[]) "If left empty (the default), the output tensors will not be written to a file.") ("event-based-profiling,e", po::bool_switch()->default_value(false), "Enables built in profiler. If unset, defaults to off.") + ("visualize-optimized-model,v", po::bool_switch()->default_value(false), + "Enables built optimized model visualizer. If unset, defaults to off.") ("fp16-turbo-mode,h", po::bool_switch()->default_value(false), "If this option is enabled, FP32 layers, " "weights and biases will be converted to FP16 where the backend supports it") ("threshold-time,r", po::value<double>(&thresholdTime)->default_value(0.0), @@ -132,6 +134,7 @@ int main(int argc, const char* argv[]) // Get the value of the switch arguments. bool concurrent = vm["concurrent"].as<bool>(); bool enableProfiling = vm["event-based-profiling"].as<bool>(); + bool enableLayerDetails = vm["visualize-optimized-model"].as<bool>(); bool enableFp16TurboMode = vm["fp16-turbo-mode"].as<bool>(); bool quantizeInput = vm["quantize-input"].as<bool>(); bool printIntermediate = vm["print-intermediate-layers"].as<bool>(); @@ -176,7 +179,8 @@ 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)); + enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, + enableLayerDetails)); } // Check results @@ -195,7 +199,8 @@ int main(int argc, const char* argv[]) { testCase.values.insert(testCase.values.begin(), executableName); if (RunCsvTest(testCase, runtime, enableProfiling, - enableFp16TurboMode, thresholdTime, printIntermediate) != EXIT_SUCCESS) + enableFp16TurboMode, thresholdTime, printIntermediate, + enableLayerDetails) != EXIT_SUCCESS) { return EXIT_FAILURE; } @@ -231,6 +236,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); + subgraphId, enableLayerDetails); } } |