From e696b6a430cd1807c73acbc4dadecfd5509c325d Mon Sep 17 00:00:00 2001
From: yangruoqi713 <yangruoqi@huawei.com>
Date: Tue, 29 Sep 2020 10:45:09 +0800
Subject: [PATCH] [MSLITE][Develop] fix bug of arm fp16 cpu op: reopen deconv
 depthwise

---
 .../runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc
index 6a9ae84eec..9509d67fb2 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc
@@ -176,6 +176,7 @@ int DeconvolutionDepthwiseFp16CPUKernel::Run() {
   }
 
   if (!need_align_) {
+    memset(execute_output_, 0, out_tensors_.at(kOutputIndex)->ElementsNum() * sizeof(float16_t));
     packed_output_ = execute_output_;
   }
   ret = ParallelLaunch(this->context_->thread_pool_, DeconvDwFp16Run, this, conv_param_->thread_num_);
@@ -243,4 +244,6 @@ kernel::LiteKernel *CpuDeconvDwFp16KernelCreator(const std::vector<lite::Tensor
   }
   return kernel;
 }
+
+REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_DeDepthwiseConv2D, CpuDeconvDwFp16KernelCreator)
 }  // namespace mindspore::kernel