diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-07-01 16:22:56 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-07-01 16:22:56 +0100 |
| commit | b17406df856d3557e6ddf7f55505359aecc061d5 (patch) | |
| tree | 653e4dc7146cb115e8e92e129dfe55b7c462fe44 /src/armnn/QuantizerVisitor.cpp | |
| parent | 970f43b078eba91c66fb64eadbc9803661ffcda8 (diff) | |
| download | armnn-b17406df856d3557e6ddf7f55505359aecc061d5.tar.gz | |
IVGCVSW-3366 Add Quantizer support for ResizeLayer
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ibfd4725bb04a5859488e968513cf11ac450fd277
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 292924c8e4..f2e0506fd6 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -385,6 +385,15 @@ void QuantizerVisitor::VisitResizeBilinearLayer(const IConnectableLayer* layer, SetQuantizedInputConnections(layer, newLayer); } +void QuantizerVisitor::VisitResizeLayer(const IConnectableLayer* layer, + const ResizeDescriptor& resizeDescriptor, + const char* name) +{ + IConnectableLayer* newLayer = m_QuantizedNetwork->AddResizeLayer(resizeDescriptor, name); + RecordLayer(layer, newLayer); + SetQuantizedInputConnections(layer, newLayer); +} + void QuantizerVisitor::VisitRsqrtLayer(const IConnectableLayer* layer, const char* name) { |