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 /1.2/HalPolicy.cpp | |
| 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 '1.2/HalPolicy.cpp')
| -rw-r--r-- | 1.2/HalPolicy.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp index c9b4dadb..8e4ef8a8 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -2697,11 +2697,11 @@ bool HalPolicy::ConvertTransposeConv2d(const Operation& operation, const Model& const uint32_t kernelX = weights.GetShape()[widthIndex]; const uint32_t kernelY = weights.GetShape()[heightIndex]; - const uint32_t outputX = outputInfo.GetShape()[widthIndex]; - const uint32_t outputY = outputInfo.GetShape()[heightIndex]; + const uint32_t outputX = outputInfo.GetShape()[widthIndex]; + const uint32_t outputY = outputInfo.GetShape()[heightIndex]; - CalcPaddingTransposeConv(outputX, kernelX, desc.m_StrideX, padLeft, padRight, paddingScheme); - CalcPaddingTransposeConv(outputY, kernelY, desc.m_StrideY, padTop, padBottom, paddingScheme); + CalcPaddingTransposeConv(outputX, kernelX, strideX, padLeft, padRight, paddingScheme); + CalcPaddingTransposeConv(outputY, kernelY, strideY, padTop, padBottom, paddingScheme); // NOTE: The Android NN API allows for negative padding values in TransposeConv2d, // but Arm NN only supports values >= 0 |