From f2e118a29519ad514b7152f3f33e2993a3ccd1e3 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9E=A5=EC=A7=80=EC=84=AD/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 15 Jan 2019 09:17:45 +0900 Subject: [PATCH] Fix infering wrong padding in TransposeConv (#4237) This commit fixes infering wrong padding in TransposeConv. Signed-off-by: jiseob.jang --- runtimes/pure_arm_compute/src/compilation.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index 323bae2..a80ab16 100644 --- a/runtimes/pure_arm_compute/src/compilation.cc +++ b/runtimes/pure_arm_compute/src/compilation.cc @@ -4052,7 +4052,7 @@ void Planner::visit(const ::internal::tflite::op::TransposeConv::Node &node) param.stride.vertical = vstride; param.padding = (padding_type == ANEURALNETWORKS_PADDING_SAME) - ? same_padding(ifm_shape, param.stride, ker_shape.W, ker_shape.H) + ? same_padding(ofm_shape, param.stride, ker_shape.W, ker_shape.H) : valid_padding(); auto stage = [param](const IAllocationContext &ctx, IExecutionBuilder &builder) { -- 2.7.4