aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2020-05-30 09:48:39 +0100
committerDavid Monahan <david.monahan@arm.com>2020-05-30 11:35:48 +0100
commit4a0c9b99deb88a0ec5de7997f09062686915c6cc (patch)
tree0456d14c0222c9a70ee1ba41f8b0ce384f4f25ad /src/armnnSerializer/Serializer.cpp
parentf540eb8111ce5d241111da487be7d817661e29b4 (diff)
downloadarmnn-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/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp8
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index c4d3cfb5dd..9c62a93e3b 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -864,7 +864,9 @@ void SerializerVisitor::VisitResizeBilinearLayer(const armnn::IConnectableLayer*
CreateResizeBilinearDescriptor(m_flatBufferBuilder,
resizeDescriptor.m_TargetWidth,
resizeDescriptor.m_TargetHeight,
- GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout));
+ GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout),
+ resizeDescriptor.m_AlignCorners,
+ resizeDescriptor.m_HalfPixelCenters);
auto flatBufferLayer = serializer::CreateResizeBilinearLayer(m_flatBufferBuilder,
flatBufferBaseLayer,
@@ -886,7 +888,9 @@ void SerializerVisitor::VisitResizeLayer(const armnn::IConnectableLayer* layer,
resizeDescriptor.m_TargetHeight,
resizeDescriptor.m_TargetWidth,
GetFlatBufferResizeMethod(resizeDescriptor.m_Method),
- GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout));
+ GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout),
+ resizeDescriptor.m_AlignCorners,
+ resizeDescriptor.m_HalfPixelCenters);
auto flatBufferLayer = serializer::CreateResizeLayer(m_flatBufferBuilder,
flatBufferBaseLayer,