From 770862800da4776ba4dcebd441e1e609ffd26d14 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Mon, 2 Sep 2019 11:46:28 +0100 Subject: IVGCVSW-2945 ExecuteNetwork should have an option to save the output to a file * Added "output-tensor-files,w" option to save output tensors to a file Signed-off-by: Sadik Armagan Change-Id: Ida457177c5cbd7dca228772405fd505d03b61bf9 --- tests/ExecuteNetwork/ExecuteNetwork.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp') diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index bccd50d929..0761551762 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -28,6 +28,7 @@ int main(int argc, const char* argv[]) std::string inputTypes; std::string outputTypes; std::string dynamicBackendsPath; + std::string outputTensorFiles; double thresholdTime = 0.0; @@ -80,6 +81,9 @@ int main(int argc, const char* argv[]) "Accepted values (float, int or qasymm8).") ("output-name,o", po::value(&outputNames), "Identifier of the output tensors in the network separated by comma.") + ("write-outputs-to-file,w", po::value(&outputTensorFiles), + "Comma-separated list of output file paths keyed with the binding-id of the output slot. " + "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.") ("fp16-turbo-mode,h", po::bool_switch()->default_value(false), "If this option is enabled, FP32 layers, " @@ -226,6 +230,7 @@ int main(int argc, const char* argv[]) return RunTest(modelFormat, inputTensorShapes, computeDevices, dynamicBackendsPath, modelPath, inputNames, inputTensorDataFilePaths, inputTypes, quantizeInput, outputTypes, outputNames, - enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, subgraphId); + outputTensorFiles, enableProfiling, enableFp16TurboMode, thresholdTime, printIntermediate, + subgraphId); } } -- cgit v1.2.1