From 1f7cf304538f52bd6fed52bbde21ea3ef7011706 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: Mon, 29 Jul 2019 20:30:43 +0900 Subject: [PATCH] Fix unmatched result of tconv in NCHW (#5968) This commit fixes unmatched result of tconv in NCHW - Change weights of tconv to OIHW in case of NCHW Signed-off-by: jiseob.jang --- runtimes/neurun/backend/acl_cl/ConstantInitializer.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtimes/neurun/backend/acl_cl/ConstantInitializer.cc b/runtimes/neurun/backend/acl_cl/ConstantInitializer.cc index 926ec09..0a8f536 100644 --- a/runtimes/neurun/backend/acl_cl/ConstantInitializer.cc +++ b/runtimes/neurun/backend/acl_cl/ConstantInitializer.cc @@ -206,7 +206,7 @@ void ConstantInitializer::visit(const model::operation::TransposeConvNode &node) { const auto &kernel_index = node.getInputs().at(model::operation::TransposeConvNode::KERNEL); const auto &kernel_obj = _operands.at(kernel_index); - registerCopyInitializer(kernel_index, kernel_obj); + registerPermuteInitializer(kernel_index, kernel_obj); } } // namespace acl_cl -- 2.7.4