Searched refs:DECONV_WINOGRAD_DEFAULT_UNIT (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 80 DECONV_WINOGRAD_DEFAULT_UNIT, unit->h_size_); in PackDeConvWgDataFp32()
448 …gradTransLeft(tile_in, at_buf, a_mid_buf, DECONV_WINOGRAD_DEFAULT_UNIT, unit_size, DECONV_WINOGRAD… in DeConvWgCalWgFp32()
450 …ogradTransRight(a_mid_buf, at_buf, trans_a_buf, unit_size, unit_size, DECONV_WINOGRAD_DEFAULT_UNIT, in DeConvWgCalWgFp32()
490 for (int hi = 0; hi < DECONV_WINOGRAD_DEFAULT_UNIT; hi++) { in DeConvWgCalCommFp32()
491 for (int wi = 0; wi < DECONV_WINOGRAD_DEFAULT_UNIT; wi++) { in DeConvWgCalCommFp32()
492 const float *src_in = tile_in + (wi + hi * DECONV_WINOGRAD_DEFAULT_UNIT) * in_stride; in DeConvWgCalCommFp32()
524 int w_start = w_unit_index * DECONV_WINOGRAD_DEFAULT_UNIT; in DeconvWg()
525 int h_start = h_unit_index * DECONV_WINOGRAD_DEFAULT_UNIT; in DeconvWg()
528 for (int hi = 0; hi < DECONV_WINOGRAD_DEFAULT_UNIT; hi++) { in DeconvWg()
529 for (int wi = 0; wi < DECONV_WINOGRAD_DEFAULT_UNIT; wi++) { in DeconvWg()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 110 deconv_param_->in_tile_w_count_ = UP_DIV(conv_param_->input_w_, DECONV_WINOGRAD_DEFAULT_UNIT); in InitParameter()
111 deconv_param_->in_tile_h_count_ = UP_DIV(conv_param_->input_h_, DECONV_WINOGRAD_DEFAULT_UNIT); in InitParameter()
121 …int size = deconv_param_->thread_num_ * DECONV_WINOGRAD_DEFAULT_UNIT * DECONV_WINOGRAD_DEFAULT_UNI… in InitParameter()
130 …deconv_param_->out_tile_w_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_w_ + conv_pa… in InitParameter()
131 …deconv_param_->out_tile_h_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_h_ + conv_pa… in InitParameter()
260 unit.winograd_.i_ = DECONV_WINOGRAD_DEFAULT_UNIT; in InitComputeParam()
261 unit.winograd_.o_ = DECONV_WINOGRAD_DEFAULT_UNIT + unit.h_size_ - 1; in InitComputeParam()
262 unit.winograd_.kh_ = unit.h_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
263 unit.winograd_.kw_ = unit.w_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
392 …float *tile_in = tile_input_ + task_id * DECONV_WINOGRAD_DEFAULT_UNIT * DECONV_WINOGRAD_DEFAULT_UN… in DoDeconv()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 123 deconv_param_->in_tile_w_count_ = UP_DIV(conv_param_->input_w_, DECONV_WINOGRAD_DEFAULT_UNIT); in InitParameter()
124 deconv_param_->in_tile_h_count_ = UP_DIV(conv_param_->input_h_, DECONV_WINOGRAD_DEFAULT_UNIT); in InitParameter()
134 …int size = deconv_param_->thread_num_ * DECONV_WINOGRAD_DEFAULT_UNIT * DECONV_WINOGRAD_DEFAULT_UNI… in InitParameter()
142 …deconv_param_->out_tile_w_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_w_ + conv_pa… in InitParameter()
143 …deconv_param_->out_tile_h_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_h_ + conv_pa… in InitParameter()
196 …float16_t *tile_in = tile_input_ + task_id * DECONV_WINOGRAD_DEFAULT_UNIT * DECONV_WINOGRAD_DEFAUL… in DoDeconv()
303 unit.winograd_.i_ = DECONV_WINOGRAD_DEFAULT_UNIT; in InitComputeParam()
304 unit.winograd_.o_ = DECONV_WINOGRAD_DEFAULT_UNIT + unit.h_size_ - 1; in InitComputeParam()
305 unit.winograd_.kh_ = unit.h_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
306 unit.winograd_.kw_ = unit.w_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 181 WinogradTransLeftFp16(tile_in, at_buf, a_mid_buf, DECONV_WINOGRAD_DEFAULT_UNIT, unit_size, in DeConvWgCalWgFp16()
182 … DECONV_WINOGRAD_DEFAULT_UNIT, deconv_param->ic_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp16()
183 …dTransRightFp16(a_mid_buf, at_buf, trans_a_buf, unit_size, unit_size, DECONV_WINOGRAD_DEFAULT_UNIT, in DeConvWgCalWgFp16()
223 for (int hi = 0; hi < DECONV_WINOGRAD_DEFAULT_UNIT; hi++) { in DeConvWgCalCommFp16()
224 for (int wi = 0; wi < DECONV_WINOGRAD_DEFAULT_UNIT; wi++) { in DeConvWgCalCommFp16()
225 const float16_t *src_in = tile_in + (wi + hi * DECONV_WINOGRAD_DEFAULT_UNIT) * in_stride; in DeConvWgCalCommFp16()
274 DECONV_WINOGRAD_DEFAULT_UNIT, unit->h_size_); in PackDeConvWgDataFp16()
354 int w_start = w_unit_index * DECONV_WINOGRAD_DEFAULT_UNIT; in DeconvWgFp16()
355 int h_start = h_unit_index * DECONV_WINOGRAD_DEFAULT_UNIT; in DeconvWgFp16()
358 for (int hi = 0; hi < DECONV_WINOGRAD_DEFAULT_UNIT; hi++) { in DeconvWgFp16()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 83 #define DECONV_WINOGRAD_DEFAULT_UNIT 3 macro
|