diff options
Diffstat (limited to 'src/armnnQuantizer/ArmNNQuantizerMain.cpp')
| -rw-r--r-- | src/armnnQuantizer/ArmNNQuantizerMain.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnnQuantizer/ArmNNQuantizerMain.cpp b/src/armnnQuantizer/ArmNNQuantizerMain.cpp index d7fc932327..9ac8966753 100644 --- a/src/armnnQuantizer/ArmNNQuantizerMain.cpp +++ b/src/armnnQuantizer/ArmNNQuantizerMain.cpp @@ -32,8 +32,14 @@ int main(int argc, char* argv[]) } } inputFileStream.close(); + + armnn::QuantizerOptions quantizerOptions; + quantizerOptions.m_ActivationFormat = cmdline.GetQuantizationScheme() == "QSymm16" + ? armnn::DataType::QuantisedSymm16 + : armnn::DataType::QuantisedAsymm8; + armnn::INetworkPtr network = parser->CreateNetworkFromBinary(binaryContent); - armnn::INetworkQuantizerPtr quantizer = armnn::INetworkQuantizer::Create(network.get()); + armnn::INetworkQuantizerPtr quantizer = armnn::INetworkQuantizer::Create(network.get(), quantizerOptions); std::string csvFileName = cmdline.GetCsvFileName(); if (csvFileName != "") |