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/armnnDeserializer/Deserializer.cpp | |
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/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index 36beebc1cd..bea34e16e7 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -2045,6 +2045,8 @@ void Deserializer::ParseResize(GraphPtr graph, unsigned int layerIndex) descriptor.m_TargetHeight = flatBufferDescriptor->targetHeight(); descriptor.m_Method = ToResizeMethod(flatBufferDescriptor->method()); descriptor.m_DataLayout = ToDataLayout(flatBufferDescriptor->dataLayout()); + descriptor.m_AlignCorners = flatBufferDescriptor->alignCorners(); + descriptor.m_HalfPixelCenters = flatBufferDescriptor->halfPixelCenters(); auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddResizeLayer(descriptor, layerName.c_str()); @@ -2073,6 +2075,8 @@ void Deserializer::ParseResizeBilinear(GraphPtr graph, unsigned int layerIndex) descriptor.m_TargetHeight = flatBufferDescriptor->targetHeight(); descriptor.m_Method = armnn::ResizeMethod::Bilinear; descriptor.m_DataLayout = ToDataLayout(flatBufferDescriptor->dataLayout()); + descriptor.m_AlignCorners = flatBufferDescriptor->alignCorners(); + descriptor.m_HalfPixelCenters = flatBufferDescriptor->halfPixelCenters(); auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddResizeLayer(descriptor, layerName.c_str()); |