diff options
Diffstat (limited to 'src/armnnQuantizer/ArmNNQuantizerMain.cpp')
-rw-r--r-- | src/armnnQuantizer/ArmNNQuantizerMain.cpp | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/src/armnnQuantizer/ArmNNQuantizerMain.cpp b/src/armnnQuantizer/ArmNNQuantizerMain.cpp index 30167e73f2..219363edbb 100644 --- a/src/armnnQuantizer/ArmNNQuantizerMain.cpp +++ b/src/armnnQuantizer/ArmNNQuantizerMain.cpp @@ -36,9 +36,19 @@ int main(int argc, char* argv[]) inputFileStream.close(); armnn::QuantizerOptions quantizerOptions; - quantizerOptions.m_ActivationFormat = cmdline.GetQuantizationScheme() == "QSymm16" - ? armnn::DataType::QSymmS16 - : armnn::DataType::QAsymmU8; + + if (cmdline.GetQuantizationScheme() == "QAsymmS8") + { + quantizerOptions.m_ActivationFormat = armnn::DataType::QAsymmS8; + } + else if (cmdline.GetQuantizationScheme() == "QSymmS16") + { + quantizerOptions.m_ActivationFormat = armnn::DataType::QSymmS16; + } + else + { + quantizerOptions.m_ActivationFormat = armnn::DataType::QAsymmU8; + } quantizerOptions.m_PreserveType = cmdline.HasPreservedDataType(); |