Fix unmatched result of tconv in NCHW (#5968)
author장지섭/On-Device Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com>
Mon, 29 Jul 2019 11:30:43 +0000 (20:30 +0900)
committer이한종/On-Device Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Mon, 29 Jul 2019 11:30:43 +0000 (20:30 +0900)
This commit fixes unmatched result of tconv in NCHW
  - Change weights of tconv to OIHW in case of NCHW

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
runtimes/neurun/backend/acl_cl/ConstantInitializer.cc

index 926ec09..0a8f536 100644 (file)
@@ -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