Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h132 int out_tile_w_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c475 … h_index * deconv_param->out_tile_w_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp32()
499 … float *dst = tile_out + h_index * out_stride * deconv_param->out_tile_w_ + w_index * out_stride; in DeConvWgCalCommFp32()
608 int merge_w_end = MSMIN(deconv_param->out_tile_w_, conv_param->output_w_ - w_start); in DeconvWgPost()
612 const float *src = src_start + (hi * deconv_param->out_tile_w_ + wi) * src_unit_stride; in DeconvWgPost()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c208 … h_index * deconv_param->out_tile_w_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp16()
232 …float16_t *dst = tile_out + h_index * out_stride * deconv_param->out_tile_w_ + w_index * out_strid… in DeConvWgCalCommFp16()
427 int merge_w_end = MSMIN(deconv_param->out_tile_w_, conv_param->output_w_ - w_start); in DeconvWgPostFp16()
431 const float16_t *src = src_start + (hi * deconv_param->out_tile_w_ + wi) * src_unit_stride; in DeconvWgPostFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc142 …deconv_param_->out_tile_w_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_w_ + conv_pa… in InitParameter()
144 size = deconv_param_->thread_num_ * deconv_param_->out_tile_w_ * deconv_param_->out_tile_h_ * in InitParameter()
198 …int size = deconv_param_->out_tile_w_ * deconv_param_->out_tile_h_ * DECONV_WINOGRAD_DEFAULT_TILE * in DoDeconv()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc130 …deconv_param_->out_tile_w_ = (DECONV_WINOGRAD_DEFAULT_UNIT - 1) * conv_param_->stride_w_ + conv_pa… in InitParameter()
394 …int size = deconv_param_->out_tile_w_ * deconv_param_->out_tile_h_ * DECONV_WINOGRAD_DEFAULT_TILE * in DoDeconv()
441 size = deconv_param_->thread_num_ * deconv_param_->out_tile_w_ * deconv_param_->out_tile_h_ * in InitRunBuf()