diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-02-18 13:25:23 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-02-20 11:52:07 +0000 |
commit | 50311ba96912125a6f6d649d0451f6fab5c1011a (patch) | |
tree | 87be8819a7b0ef13428dbade1b7cce528d1060d4 /tests/ExecuteNetwork/ExecuteNetwork.cpp | |
parent | 67e6c54b4c4ff1e46a9ed14014d279794a666969 (diff) | |
download | armnn-50311ba96912125a6f6d649d0451f6fab5c1011a.tar.gz |
COMPMID-3062: Add ExecuteNetwork option to explicitly dequantize output
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ic7abe5dad700fbdc1b7cbab6936b52554f0a0360
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 55864c8e02..3c0301010d 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -83,6 +83,10 @@ int main(int argc, const char* argv[]) "The type of the output tensors in the network separated by comma. " "If unset, defaults to \"float\" for all defined outputs. " "Accepted values (float, int or qasymm8).") + ("dequantize-output,l",po::bool_switch()->default_value(false), + "If this option is enabled, all quantized outputs will be dequantized to float. " + "If unset, default to not get dequantized. " + "Accepted values (true or false)") ("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), @@ -153,6 +157,7 @@ int main(int argc, const char* argv[]) bool enableLayerDetails = vm["visualize-optimized-model"].as<bool>(); bool enableFp16TurboMode = vm["fp16-turbo-mode"].as<bool>(); bool quantizeInput = vm["quantize-input"].as<bool>(); + bool dequantizeOutput = vm["dequantize-output"].as<bool>(); 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>(); @@ -270,7 +275,7 @@ 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, parseUnsupported, runtime); + outputTensorFiles, dequantizeOutput, enableProfiling, enableFp16TurboMode, thresholdTime, + printIntermediate, subgraphId, enableLayerDetails, parseUnsupported, runtime); } } |