Home
last modified time | relevance | path

Searched refs:ic_up4_ (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c144 size = deconv_param->ic_up4_ * deconv_param->oc_up4_ * tmp_kernel_plane; in PackDeConvWgDataFp32()
151 …index = upi * deconv_param->oc_up4_ * deconv_param->ic_up4_ + oc4div * C4NUM * deconv_param->ic_up… in PackDeConvWgDataFp32()
456 float *src = trans_a_buf + index * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeConvWgCalWgFp32()
458 const float *weight = weight_buf + index * deconv_param->ic_up4_ * deconv_param->oc_up4_; in DeConvWgCalWgFp32()
487 int in_stride = DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeConvWgCalCommFp32()
516 int tile_in_unit_stride = deconv_param->ic_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeconvWg()
566 … DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeconvWg()
568 … DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c189 float16_t *src = trans_a_buf + index * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeConvWgCalWgFp16()
191 const float16_t *weight = weight_buf + index * deconv_param->ic_up4_ * deconv_param->oc_up4_; in DeConvWgCalWgFp16()
220 int in_stride = DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeConvWgCalCommFp16()
325 size = deconv_param->ic_up4_ * deconv_param->oc_up4_ * tmp_kernel_plane; in PackDeConvWgDataFp16()
332 …index = upi * deconv_param->oc_up4_ * deconv_param->ic_up4_ + oc4div * C4NUM * deconv_param->ic_up… in PackDeConvWgDataFp16()
347 int tile_in_unit_stride = deconv_param->ic_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeconvWgFp16()
387 … DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeconvWgFp16()
389 … DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->ic_up4_; in DeconvWgFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc122 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in InitParameter()
143 … = unit.winograd_.kh_ * unit.winograd_.kw_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in InitParameter()
215 deconv_param_->ic_up4_ = deconv_param_->ic_div4_ * C4NUM; in InitComputeParam()
267 deconv_param_->ic_up4_ * sizeof(float)); in InitComputeParam()
274 …unit.weight_ = malloc(h_size * w_size * deconv_param_->ic_up4_ * deconv_param_->oc_up4_ * sizeof(f… in InitComputeParam()
393 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in DoDeconv()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc135 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in InitParameter()
160 … = unit.winograd_.kh_ * unit.winograd_.kw_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in InitParameter()
197 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->ic_up4_; in DoDeconv()
262 deconv_param_->ic_up4_ = deconv_param_->ic_div4_ * C4NUM; in InitComputeParam()
310 deconv_param_->ic_up4_ * sizeof(float16_t)); in InitComputeParam()
316 …unit.weight_ = malloc(h_size * w_size * deconv_param_->ic_up4_ * deconv_param_->oc_up4_ * sizeof(f… in InitComputeParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h125 int ic_up4_; member