diff options
author | Mike Kelly <mike.kelly@arm.com> | 2020-01-15 10:02:33 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-01-28 17:03:27 +0000 |
commit | 26123db75b77b4bba9878ef0eb1c9d7d2651ccee (patch) | |
tree | 9f6168de146b99f22d8f7e86be6f36ce61f25211 /ConversionUtils.hpp | |
parent | 370b9b73f59adae331cccfc117caef04a028333a (diff) | |
download | android-nn-driver-26123db75b77b4bba9878ef0eb1c9d7d2651ccee.tar.gz |
MLCE-153 Padding calculation for TransposeConv ignores stride
* Changed ConvertTransposeConv2d to use the unsigned values for strideX and strideY
not the signed values from the descriptor.
!ComputeLibrary:2599
!armnn:2612
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: Iee300041f3ed585820b7c7e2edd8ab541f339d59
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 4881ec9e..550c3201 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -358,7 +358,7 @@ void CalcPadding(uint32_t input, uint32_t kernel, uint32_t stride, uint32_t dila outPadTail = boost::numeric_cast<uint32_t>(padTail); } -void CalcPaddingTransposeConv(uint32_t output, uint32_t kernel, uint32_t stride, int32_t& outPadHead, +void CalcPaddingTransposeConv(uint32_t output, uint32_t kernel, int32_t stride, int32_t& outPadHead, int32_t& outPadTail, android::nn::PaddingScheme scheme) { calculateExplicitPaddingTransposeConv(output, stride, kernel, scheme, &outPadHead, &outPadTail); |