diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-09-02 11:46:28 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-09-02 11:46:28 +0100 |
commit | 770862800da4776ba4dcebd441e1e609ffd26d14 (patch) | |
tree | 92b158a0761893808d95c12b10d87bad148eeb45 /tests/ExecuteNetwork | |
parent | 56e26ba60a7a42c657b1d36a38d7b766cf51e452 (diff) | |
download | armnn-770862800da4776ba4dcebd441e1e609ffd26d14.tar.gz |
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 <sadik.armagan@arm.com>
Change-Id: Ida457177c5cbd7dca228772405fd505d03b61bf9
Diffstat (limited to 'tests/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
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); } } |