Browse Source

!6270 [MS][LITE][Develop] enable DeConvFp32Center define

Merge pull request !6270 from liuzhongkai/arm32_new1
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
7ee905b977
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/nnacl/fp32/conv_depthwise.c

+ 1
- 1
mindspore/lite/nnacl/fp32/conv_depthwise.c View File

@@ -440,7 +440,7 @@ void DeconvDwC4Fp32(float *output_data, const float *input_data, const float *we
float *out_t = dst_data + oh_h_start * sliding->in_h_step_ + oh_w_start * sliding->block_channel_;
const float *in_t = src_data + sliding->top_ * sliding->out_h_step_ + sliding->left_ * sliding->block_channel_;

#ifdef ENABLE_ARM64
#ifdef ENABLE_ARM
DeconvDwFp32Center(out_t, in_t, weight, sliding->bottom_ - sliding->top_, sliding->right_ - sliding->left_,
conv_param->kernel_h_, conv_param->kernel_w_, sliding->out_h_step_ * sizeof(float),
sliding->block_channel_ * sizeof(float), sliding->in_sh_step_ * sizeof(float),


Loading…
Cancel
Save