diff options
author | Kevin May <kevin.may@arm.com> | 2020-06-03 16:05:00 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2020-06-03 15:35:26 +0000 |
commit | 1c6e9763aaac923660b795a55544b50f4b179d77 (patch) | |
tree | 9c5d5a9c9437192f4438594e45f6d73639cd0923 /python/pyarmnn | |
parent | 0dbe00a9c6fdfc20a3d7815b05269da0e8df7b0a (diff) | |
download | armnn-1c6e9763aaac923660b795a55544b50f4b179d77.tar.gz |
IVGCVSW-4774 Fix resize descriptor member name change for Pyarmnn
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I4e7c8341200d58ab8ae3c574ff622278d3938f95
Diffstat (limited to 'python/pyarmnn')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i | 4 | ||||
-rw-r--r-- | python/pyarmnn/test/test_descriptors.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i index 75b875e638..9a01a52e13 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i @@ -660,7 +660,7 @@ struct ReshapeDescriptor m_Method (int): The Interpolation method to use (`ResizeMethod_Bilinear`, `ResizeMethod_NearestNeighbor`). Default: `ResizeMethod_NearestNeighbor`. m_DataLayout (int): The data layout to be used (`DataLayout_NCHW`, `DataLayout_NHWC`). Default: `DataLayout_NCHW`. - m_BilinearAlignCorners (bool): Align corners or not when resizing bilinearly. If True, corner pixel values are preserved after resizing. + m_AlignCorners (bool): Align corners or not when resizing. If True, corner pixel values are preserved after resizing. Default: False ") ResizeDescriptor; @@ -672,7 +672,7 @@ struct ResizeDescriptor uint32_t m_TargetHeight; ResizeMethod m_Method; DataLayout m_DataLayout; - bool m_BilinearAlignCorners; + bool m_AlignCorners; bool operator ==(const ResizeDescriptor& rhs) const; }; diff --git a/python/pyarmnn/test/test_descriptors.py b/python/pyarmnn/test/test_descriptors.py index 1a41105aa9..6d49747d5a 100644 --- a/python/pyarmnn/test/test_descriptors.py +++ b/python/pyarmnn/test/test_descriptors.py @@ -270,7 +270,7 @@ def test_resize_descriptor_default_values(): assert desc.m_TargetHeight == 0 assert desc.m_Method == ann.ResizeMethod_NearestNeighbor assert desc.m_DataLayout == ann.DataLayout_NCHW - assert desc.m_BilinearAlignCorners == False + assert desc.m_AlignCorners == False def test_spacetobatchnd_descriptor_default_values(): |