diff options
| author | David Monahan <david.monahan@arm.com> | 2020-05-30 09:48:39 +0100 |
|---|---|---|
| committer | David Monahan <david.monahan@arm.com> | 2020-05-30 11:35:48 +0100 |
| commit | 4a0c9b99deb88a0ec5de7997f09062686915c6cc (patch) | |
| tree | 0456d14c0222c9a70ee1ba41f8b0ce384f4f25ad /src/armnn | |
| parent | f540eb8111ce5d241111da487be7d817661e29b4 (diff) | |
| download | armnn-4a0c9b99deb88a0ec5de7997f09062686915c6cc.tar.gz | |
IVGCVSW-4888 Update Resize Front end for new parameters
* Added AlignCorners and HalfPixelCenters to Resize and ResizeBilinear
* Updated Serializer and Serializer tests
!android-nn-driver:3280
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: Ice3bb448c239b04323854641525bfe0808c03b2e
Diffstat (limited to 'src/armnn')
| -rw-r--r-- | src/armnn/Network.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index a047297e15..d636d01291 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1458,10 +1458,12 @@ IConnectableLayer* Network::AddResizeBilinearLayer(const ResizeBilinearDescripto const char* name) { ResizeDescriptor resizeDescriptor; - resizeDescriptor.m_Method = ResizeMethod::Bilinear; - resizeDescriptor.m_DataLayout = descriptor.m_DataLayout; - resizeDescriptor.m_TargetWidth = descriptor.m_TargetWidth; - resizeDescriptor.m_TargetHeight = descriptor.m_TargetHeight; + resizeDescriptor.m_Method = ResizeMethod::Bilinear; + resizeDescriptor.m_DataLayout = descriptor.m_DataLayout; + resizeDescriptor.m_TargetWidth = descriptor.m_TargetWidth; + resizeDescriptor.m_TargetHeight = descriptor.m_TargetHeight; + resizeDescriptor.m_AlignCorners = descriptor.m_AlignCorners; + resizeDescriptor.m_HalfPixelCenters = descriptor.m_HalfPixelCenters; return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name); } |