From c8bdb3943635a7beb0fdc8538803c8e61a3abe33 Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Thu, 1 Aug 2019 15:51:44 +0100
Subject: 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
---
 ConversionUtils.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'ConversionUtils.hpp')

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)
-- 
cgit v1.2.1