Searched refs:w_start_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 108 int w_start_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 66 int src_w_offset = unit->w_start_ + uwi * conv_param->stride_w_; in PackDeConvWgDataFp32()
572 … wg_dst_a_buf, transferred, unit->winograd_.BT_, tmp_b_buf, unit->winograd_.kh_, unit->w_start_, in DeconvWg()
577 …nvWgCalCommFp32(tile_in, tile_out, (float *)unit->weight_, tmp_buf, unit->h_start_, unit->w_start_, in DeconvWg()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 257 int src_w_offset = unit->w_start_ + uwi * conv_param->stride_w_; in PackDeConvWgDataFp16()
394 … transferred, unit->winograd_.BT_, tmp_b, unit->winograd_.kh_, unit->w_start_, unit->h_start_, in DeconvWgFp16()
399 …CalCommFp16(tile_in, tile_out, (float16_t *)unit->weight_, tmp_buf, unit->h_start_, unit->w_start_, in DeconvWgFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 295 unit.w_start_ = si_w; in InitComputeParam()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 252 unit.w_start_ = si_w; in InitComputeParam()
|