diff options
-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(): |