diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-01 19:01:44 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-07-02 13:50:24 +0000 |
| commit | 169d2f120cc9021f170fede22a448fd6b66fc979 (patch) | |
| tree | a55b62678a85f63c074148d33fae744b22c834a8 /src/armnn/QuantizerVisitor.cpp | |
| parent | 38e05bd2836b1b65b440330a9c283038ba4192c3 (diff) | |
| download | armnn-169d2f120cc9021f170fede22a448fd6b66fc979.tar.gz | |
IVGCVSW-3382 Deprecate ResizeBilinear and use Resize with Bilinear method
!android-nn-driver:1451
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ieedbce1f6e95891137a250fdd07e2f7e4e1f4828
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
| -rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index f2e0506fd6..37c254102a 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -377,12 +377,16 @@ void QuantizerVisitor::VisitReshapeLayer(const IConnectableLayer* layer, } void QuantizerVisitor::VisitResizeBilinearLayer(const IConnectableLayer* layer, - const ResizeBilinearDescriptor& resizeDesc, + const ResizeBilinearDescriptor& resizeBilinearDescriptor, const char* name) { - IConnectableLayer* newLayer = m_QuantizedNetwork->AddResizeBilinearLayer(resizeDesc, name); - RecordLayer(layer, newLayer); - SetQuantizedInputConnections(layer, newLayer); + ResizeDescriptor resizeDescriptor; + resizeDescriptor.m_Method = ResizeMethod::Bilinear; + resizeDescriptor.m_TargetWidth = resizeBilinearDescriptor.m_TargetWidth; + resizeDescriptor.m_TargetHeight = resizeBilinearDescriptor.m_TargetHeight; + resizeDescriptor.m_DataLayout = resizeBilinearDescriptor.m_DataLayout; + + VisitResizeLayer(layer, resizeDescriptor, name); } void QuantizerVisitor::VisitResizeLayer(const IConnectableLayer* layer, |