diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-05-09 10:13:20 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-10 08:30:55 +0000 |
| commit | 724e48013142562b7f09c9c819f57c314c4ee3d4 (patch) | |
| tree | 935b425961ccb13a488708312bcc70b3b32fc87e /src/armnn/NetworkQuantizer.cpp | |
| parent | 5fa83938592db420914903235daf3f1d5c97d6bc (diff) | |
| download | armnn-724e48013142562b7f09c9c819f57c314c4ee3d4.tar.gz | |
IVGCVSW-3061 Modify NetworkQuantizer to support option to preserve input/output types
* Also add unit tests for new preserve type option
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Change-Id: I860759072f2e3546698118d1bcd5e79eb4e805ec
Diffstat (limited to 'src/armnn/NetworkQuantizer.cpp')
| -rw-r--r-- | src/armnn/NetworkQuantizer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/NetworkQuantizer.cpp b/src/armnn/NetworkQuantizer.cpp index 12e459d276..f308d54d49 100644 --- a/src/armnn/NetworkQuantizer.cpp +++ b/src/armnn/NetworkQuantizer.cpp @@ -171,7 +171,7 @@ INetworkPtr NetworkQuantizer::ExportNetwork() throw InvalidArgumentException("Unsupported quantization target"); } - QuantizerVisitor quantizerVisitor(m_Ranges, quantizationScheme.get()); + QuantizerVisitor quantizerVisitor(m_Ranges, quantizationScheme.get(), m_Options.m_PreserveType); VisitLayers(graph, quantizerVisitor); // clear the ranges |