diff options
Diffstat (limited to 'src/armnn/NetworkQuantizer.cpp')
| -rw-r--r-- | src/armnn/NetworkQuantizer.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/NetworkQuantizer.cpp b/src/armnn/NetworkQuantizer.cpp index 5e00de5fcb..d55fca68b5 100644 --- a/src/armnn/NetworkQuantizer.cpp +++ b/src/armnn/NetworkQuantizer.cpp @@ -161,7 +161,10 @@ INetworkPtr NetworkQuantizer::ExportNetwork() switch (m_Options.m_ActivationFormat) { case DataType::QAsymmU8: - quantizationScheme = std::make_unique<QAsymm8QuantizationScheme>(); + quantizationScheme = std::make_unique<QAsymmU8QuantizationScheme>(); + break; + case DataType::QAsymmS8: + quantizationScheme = std::make_unique<QAsymmS8QuantizationScheme>(); break; case DataType::QSymmS8: quantizationScheme = std::make_unique<QSymmS8QuantizationScheme>(); |