diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-02-18 10:42:32 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2019-02-18 10:42:32 +0000 |
commit | 2ee88dfe7096f8f571ae7be9cbf0f49ededd89af (patch) | |
tree | 85f3188cdf268842d9dc293bb1aced01188d6016 /src/armnn/QuantizerVisitor.cpp | |
parent | aa80d8a9f334254188aa244d3a438846f3996177 (diff) | |
download | armnn-2ee88dfe7096f8f571ae7be9cbf0f49ededd89af.tar.gz |
IVGCVSW-2626 Add Quantization of ResizeBilinear Layer
Change-Id: I0daffd23f32cc094d8309aed822cc46bb2aaa46f
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index 2ca164bf36..8b009c679f 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -280,4 +280,13 @@ void QuantizerVisitor::VisitReshapeLayer(const IConnectableLayer* layer, SetQuantizedInputConnections(layer, newLayer); } +void QuantizerVisitor::VisitResizeBilinearLayer(const IConnectableLayer* layer, + const ResizeBilinearDescriptor& resizeDesc, + const char* name) +{ + IConnectableLayer* newLayer = m_QuantizedNetwork->AddResizeBilinearLayer(resizeDesc, name); + RecordLayer(layer, newLayer); + SetQuantizedInputConnections(layer, newLayer); +} + } //namespace armnn |