diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2020-03-10 13:51:45 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-03-10 14:02:00 +0000 |
commit | ddb1d06dbcb5dc4a89a237ac1176279669817f46 (patch) | |
tree | e2d1104bea631f8f366f529201041e701d09b998 /src/armnnQuantizer/ArmNNQuantizerMain.cpp | |
parent | 6445cfff7519effd1df04eac88ae17d6e4e6693b (diff) | |
download | armnn-ddb1d06dbcb5dc4a89a237ac1176279669817f46.tar.gz |
MLCE-159 Add QAsymmS8 to ArmnnQuantizer
* Allow per layer quantization from Fp32 to Int8 (QAsymmS8) like TfLite
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I5bbf770aa29d81af3568c15b47d2b2c18e55bb28
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(); |