diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2019-04-26 17:11:47 +0100 |
|---|---|---|
| committer | Les Bell <les.bell@arm.com> | 2019-04-29 12:06:53 +0000 |
| commit | dc2f7f4bd79e59e0574461b563c77d4b1934aa09 (patch) | |
| tree | bb0067315b071fda97407b3b68c1ff176bd046e5 /src/armnnQuantizer/ArmNNQuantizerMain.cpp | |
| parent | ae2c5f0350a7033f58578f9c509345445a639865 (diff) | |
| download | armnn-dc2f7f4bd79e59e0574461b563c77d4b1934aa09.tar.gz | |
IVGCVSW-2843 Support QSymm16 via Command-line quantizer tool
* Added quantization scheme support to ArmnnQuantizer
Change-Id: Ia0202bfcd8cac0695aa3b18f9447a19b27f74f67
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
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 != "") |