| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | depthwise_conv2d_infer.c | 47 param->kernel_w_ = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(inputs[kWeightIndex]); in DepthwiseConv2dInferShape()
52 …int pad_w_all = ((output_w - 1) * param->stride_w_ + (param->kernel_w_ - 1) * param->dilation_w_ +… in DepthwiseConv2dInferShape()
66 ((float)(param->kernel_w_) - 1) * (float)(param->dilation_w_)) / in DepthwiseConv2dInferShape()
|
| D | conv2d_infer.c | 20 int kernel_w = param->kernel_w_; in ConvInferShape()
63 param->kernel_w_ > input_w + param->pad_l_ + param->pad_r_) { in ConvInferShape()
114 param->kernel_w_ = param->kernel_w_ != -1 ? param->kernel_w_ : weight_tensor->shape_[2]; in Conv2dInferShape()
|
| D | deconv2d_infer.c | 54 int kernel_w = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(weight); in Deconv2dInferShape()
111 param->kernel_w_ = kernel_w; in Deconv2dInferShape()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | deconvolution_int8_coder.cc | 55 …tmul_param_->col_ = conv_param_->output_channel_ * conv_param_->kernel_h_ * conv_param_->kernel_w_; in InitParam()
78 conv_param_->kernel_w_ * conv_param_->kernel_h_ * sizeof(int8_t); in InitBiasWeight()
85 … conv_param_->output_channel_, conv_param_->kernel_h_ * conv_param_->kernel_w_, in InitBiasWeight()
88 … = UP_ROUND(conv_param_->output_channel_, C4NUM) * conv_param_->kernel_h_ * conv_param_->kernel_w_; in InitBiasWeight()
110 … UP_ROUND(conv_param_->output_channel_, C4NUM) * conv_param_->kernel_w_ * conv_param_->kernel_h_ * in InitRunBuf()
150 int kernel_plane = conv_param_->kernel_w_ * conv_param_->kernel_h_; in DoCode()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | depthwise_conv2d_infer_test.cc | 46 parameter->kernel_w_ = 3; in TEST_F()
92 parameter->kernel_w_ = 7; in TEST_F()
138 parameter->kernel_w_ = 3; in TEST_F()
185 parameter->kernel_w_ = 3; in TEST_F()
232 parameter->kernel_w_ = 5; in TEST_F()
279 parameter->kernel_w_ = 3; in TEST_F()
326 parameter->kernel_w_ = 1; in TEST_F()
373 parameter->kernel_w_ = 1; in TEST_F()
420 parameter->kernel_w_ = 3; in TEST_F()
467 parameter->kernel_w_ = 1; in TEST_F()
[all …]
|
| D | conv2d_infer_test.cc | 45 parameter->kernel_w_ = 3; in TEST_F()
90 parameter->kernel_w_ = 7; in TEST_F()
135 parameter->kernel_w_ = 3; in TEST_F()
181 parameter->kernel_w_ = 3; in TEST_F()
227 parameter->kernel_w_ = 5; in TEST_F()
273 parameter->kernel_w_ = 3; in TEST_F()
319 parameter->kernel_w_ = 1; in TEST_F()
365 parameter->kernel_w_ = 1; in TEST_F()
411 parameter->kernel_w_ = 3; in TEST_F()
457 parameter->kernel_w_ = 1; in TEST_F()
[all …]
|
| D | deconv2d_infer_test.cc | 47 parameter->kernel_w_ = 3; in TEST_F()
95 parameter->kernel_w_ = 3; in TEST_F()
143 parameter->kernel_w_ = 2; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | conv_common_base.c | 24 return conv_param->kernel_h_ == 3 && conv_param->kernel_w_ == 3 && conv_param->stride_w_ == 1 && in CheckConvDw1DWinograd()
47 int kernel_w = conv_param->kernel_w_; in SelectOutputUnit()
86 …if (conv_param->kernel_w_ == conv_param->kernel_h_ && conv_param->dilation_h_ == 1 && conv_param->… in CheckIfUseWinograd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 63 … const float *weight_kh = weight_data + kh * conv_param->kernel_w_ * conv_param->output_channel_; in ConvDw()
66 for (int kw = 0; kw < conv_param->kernel_w_; kw++) { in ConvDw()
106 …while ((right - 1) * conv_param->stride_w_ - conv_param->pad_l_ + conv_param->kernel_w_ * conv_par… in InitSlidingParam()
160 …sliding->kernel_step_ = conv_param->kernel_w_ * conv_param->kernel_h_ * sliding->ic_align_ * weigh… in AppendSlidingParamConv()
175 sliding->kernel_step_ = conv_param->kernel_w_ * conv_param->kernel_h_ * block; in AppendSlidingParamConvDw()
224 …int end_kw = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->input_w_ - iw, conv_param->dilation_w… in ConvDwBorder()
228 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in ConvDwBorder()
242 param->kernel_w = conv_param->kernel_w_ * C4NUM * sizeof(float); in ConvDwBorder()
250 conv_param->kernel_w_ * C4NUM * sizeof(float), relu, relu6); in ConvDwBorder()
253 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_ * C4NUM, relu, relu6); in ConvDwBorder()
[all …]
|
| D | deconv_fp32.c | 41 size_t kernel_plane = conv_param->kernel_w_ * conv_param->kernel_h_; in DeConvPostFp32C8()
53 int src_kh_stride = in_plane_round * conv_param->kernel_w_ * C8NUM; in DeConvPostFp32C8()
74 …int kw_end = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeConvPostFp32C8()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | deconvolution_grad_fp32_tests.cc | 48 conv_param->kernel_w_ = 3; in TEST_F()
74 …conv_param->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_ch… in TEST_F()
156 conv_param->kernel_w_ = 3; in TEST_F()
182 …conv_param->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_ch… in TEST_F()
258 conv_param->kernel_w_ = 3; in TEST_F()
287 …conv_param->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_ch… in TEST_F()
365 conv_param->kernel_w_ = 3; in TEST_F()
391 …conv_param->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_ch… in TEST_F()
472 conv_param->kernel_w_ = 3; in TEST_F()
498 …conv_param->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_ch… in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_fp16.c | 23 size_t kernel_plane = conv_param->kernel_w_ * conv_param->kernel_h_; in DeConvPostFp16()
30 int src_kh_stride = in_plane16 * conv_param->kernel_w_ * C8NUM; in DeConvPostFp16()
49 …int kw_end = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeConvPostFp16()
|
| D | conv_depthwise_fp16.c | 423 …const float16_t *weight_kh = weight_data + kh * conv_param->kernel_w_ * conv_param->output_channel… in ConvDwFp16()
426 for (int kw = 0; kw < conv_param->kernel_w_; kw++) { in ConvDwFp16()
500 …int end_kw = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->input_w_ - iw, conv_param->dilation_w… in DepthwiseBorderFp16()
504 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DepthwiseBorderFp16()
508 conv_param->kernel_w_ * C8NUM * sizeof(float16_t), relu, relu6); in DepthwiseBorderFp16()
511 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_ * C8NUM, relu, relu6); in DepthwiseBorderFp16()
602 … conv_param->kernel_h_, conv_param->kernel_w_, sliding->out_h_step_ * sizeof(float16_t), in ConvDwC8Fp16()
608 … sliding->right_ - sliding->left_, conv_param->kernel_h_, conv_param->kernel_w_, in ConvDwC8Fp16()
658 …int end_kw = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeconvDepthwiseBorderFp16()
661 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DeconvDepthwiseBorderFp16()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | deconv2d_populate.cc | 47 param->kernel_w_ = -1; in PopulateDeconvParameter()
55 param->kernel_w_ = static_cast<int>(*(kernel_size->begin() + 1)); in PopulateDeconvParameter()
79 param->kernel_w_ = static_cast<int>(*(kernel_size->begin() + 1)); in PopulateDeconvParameter()
|
| D | conv2d_populate.cc | 51 param->kernel_w_ = static_cast<int>(*(kernel_size->begin() + 1)); in PopulateConvParameter()
54 param->kernel_w_ = -1; in PopulateConvParameter()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | convolution.cc | 51 conv_param_->kernel_w_ = input_weight->shape().at(kNHWC_W); in ReSize()
55 …const int k = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ / conv… in ReSize()
57 …do_img2col_ = (conv_param_->kernel_h_ == 1) && (conv_param_->kernel_w_ == 1) && (conv_param_->pad_… in ReSize()
69 ws_size_ = chunk_ * conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_; in ReSize()
94 const int k_w = conv_param_->kernel_w_; in Execute()
|
| D | convolution_grad_input.cc | 53 …ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / c… in ReSize()
55 …int n = conv_param->kernel_w_ * conv_param->kernel_h_ * conv_param->input_channel_ / conv_param->g… in ReSize()
61 …do_img2col_ = (conv_param->kernel_h_ == 1) && (conv_param->kernel_w_ == 1) && (conv_param->pad_d_ … in ReSize()
95 int k_w = conv_param->kernel_w_; in Execute()
|
| D | deconvolution_grad_filter.cc | 59 …ws_size = chunk * conv_param->input_w_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_para… in Init()
63 …int n = conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->output_channel_ / conv_param->… in Init()
91 int k_w = conv_param->kernel_w_; in Execute()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_fp32.cc | 61 auto kernel_w_ = weight_tensor->Width(); in MallocWeightBiasData() local
63 …size_t pack_weight_size = input_channel * kernel_w_ * kernel_h_ * output_aligned_size * sizeof(flo… in MallocWeightBiasData()
100 kernel_plane_ = conv_param_->kernel_w_ * conv_param_->kernel_h_; in InitParam()
181 auto kernel_w_ = weight_tensor->Width(); in Init() local
183 …size_t pack_weight_size = input_channel * kernel_w_ * kernel_h_ * output_aligned_size * sizeof(flo… in Init()
302 …(conv_param->kernel_w_ / conv_param->stride_w_ >= 2 || conv_param->kernel_h_ / conv_param->stride_… in CpuDeConvFp32KernelCreator()
|
| D | convolution_slidewindow_fp32.cc | 34 if (conv_param_->kernel_h_ == 1 && conv_param_->kernel_w_ == 1) { in Init()
60 if (!(conv_param_->kernel_h_ == 1 && conv_param_->kernel_w_ == 1)) { in Init()
94 if (conv_param_->kernel_w_ == 1 && conv_param_->kernel_h_ == 1) { in RunImpl()
118 if (ic_res_ != 0 && conv_param_->kernel_h_ == 1 && conv_param_->kernel_w_ == 1) { in InitTmpBuffer()
|
| D | deconvolution_winograd_fp32.cc | 130 …out_tile_w_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_w_ + conv_param_->kernel_w_; in InitParameter()
209 conv_param_->kernel_w_ = weight_tensor->Width(); in InitComputeParam()
212 deconv_param_->kernel_plane_ = conv_param_->kernel_w_ * conv_param_->kernel_h_; in InitComputeParam()
221 if (si_h < conv_param_->kernel_h_ && si_w < conv_param_->kernel_w_) { in InitComputeParam()
242 if (si_w >= conv_param_->kernel_w_) { in InitComputeParam()
247 int w_size = 1 + (conv_param_->kernel_w_ - si_w - 1) / conv_param_->stride_w_; in InitComputeParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pack_ext.c | 33 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp32()
83 const int kernel_w = conv_param->kernel_w_; in rolling_im2col_hwc()
154 const int kernel_w = conv_param->kernel_w_; in rolling_im2row_hwc()
202 const int kernel_w = conv_param->kernel_w_; in col2im_hwc()
250 const int kernel_w = conv_param->kernel_w_; in rolling_col2im_hwc()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | convolution_fp16_grad_input.cc | 53 …ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / c… in ReSize()
55 …int n = conv_param->kernel_w_ * conv_param->kernel_h_ * conv_param->input_channel_ / conv_param->g… in ReSize()
61 …do_img2col_ = (conv_param->kernel_h_ == 1) && (conv_param->kernel_w_ == 1) && (conv_param->pad_d_ … in ReSize()
94 int k_w = conv_param->kernel_w_; // out_dw->shape()[2]; in DoExecute()
|
| D | convolution_fp16_grad_filter.cc | 57 …do_img2col_ = (conv_param->kernel_h_ == 1) && (conv_param->kernel_w_ == 1) && (conv_param->pad_d_ … in ReSize()
69 ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in ReSize()
71 …int n = conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / conv_param->g… in ReSize()
98 int k_w = conv_param->kernel_w_; in DoExecute()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | pack_tests.cc | 45 conv_param->kernel_w_ = 3; in InitConvParamPack()
64 int kernel_w = conv_param->kernel_w_; in TEST_F()
127 int kernel_w = conv_param->kernel_w_; in TEST_F()
|