Searched refs:ic_div4_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 449 deconv_param->ic_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp32()
451 deconv_param->ic_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp32()
459 … TiledC4MatmulFp32(dst, src, weight, DECONV_WINOGRAD_DEFAULT_TILE * C4NUM, deconv_param->ic_div4_, in DeConvWgCalWgFp32()
493 …tmulFp32(tmp_buf, src_in, weight, DECONV_WINOGRAD_DEFAULT_TILE * 4, deconv_param->ic_div4_, count); in DeConvWgCalCommFp32()
534 for (int ic4_index = 0; ic4_index < deconv_param->ic_div4_; ic4_index++) { in DeconvWg()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 182 … DECONV_WINOGRAD_DEFAULT_UNIT, deconv_param->ic_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp16()
184 deconv_param->ic_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp16()
192 … TiledC4MatmulFp16(dst, src, weight, DECONV_WINOGRAD_DEFAULT_TILE * C4NUM, deconv_param->ic_div4_, in DeConvWgCalWgFp16()
226 …tmulFp16(tmp_buf, src_in, weight, DECONV_WINOGRAD_DEFAULT_TILE * 4, deconv_param->ic_div4_, count); in DeConvWgCalCommFp16()
364 for (int ic4_index = 0; ic4_index < deconv_param->ic_div4_; ic4_index++) { in DeconvWgFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 123 int ic_div4_; member
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 213 deconv_param_->ic_div4_ = UP_DIV(conv_param_->input_channel_, C4NUM); in InitComputeParam()
215 deconv_param_->ic_up4_ = deconv_param_->ic_div4_ * C4NUM; in InitComputeParam()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 260 deconv_param_->ic_div4_ = UP_DIV(conv_param_->input_channel_, C4NUM); in InitComputeParam()
262 deconv_param_->ic_up4_ = deconv_param_->ic_div4_ * C4NUM; in InitComputeParam()
|