diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-01 15:51:44 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-02 14:24:00 +0000 |
commit | c8bdb3943635a7beb0fdc8538803c8e61a3abe33 (patch) | |
tree | 691ae75b3a7160be7f98bc839846523287e9a7a4 /ConversionUtils.hpp | |
parent | f094dba07b1e1dd65d8b85adcbbd98f2d77cb07a (diff) | |
download | android-nn-driver-c8bdb3943635a7beb0fdc8538803c8e61a3abe33.tar.gz |
IVGCVSW-3604 Fix TransposeConv2d padding calculation
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Signed-off-by: Aron Virginas-Tar <aron.virginas-tar@arm.com>
Change-Id: I5c10ab18343ecf0ebeab24a436e5be2b6c2831c7
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 18a65413..9471d781 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -323,6 +323,12 @@ 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, + int32_t& outPadTail, android::nn::PaddingScheme scheme) +{ + calculateExplicitPaddingTransposeConv(output, stride, kernel, scheme, &outPadHead, &outPadTail); +} + #endif Shape GetOperandShape(const V1_0::Operand& operand) |