| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_common_fp32.h | 47 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
62 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
68 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
73 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
78 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
83 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
88 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
93 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
98 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
103 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
[all …]
|
| D | conv_depthwise_fp32.h | 68 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
72 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
76 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
80 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
84 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
88 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
92 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
96 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
100 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
117 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
|
| D | conv_depthwise_fp32.c | 1069 int oc_algin = sw_param->block_channel_; in DepthwiseSWAvxFp32() local
1085 int in_start = in_h_start * sw_param->in_h_step_ + in_w_start * oc_algin; in DepthwiseSWAvxFp32()
1121 …dst_w + ow * oc_algin, src_w, weight, bias, kernel_h, kernel_w, act_type, ow_block, oc_block, oc_a… in DepthwiseSWAvxFp32()
1138 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSWWxKKernel() argument
1192 _mm256_storeu_ps(dst + i * oc_algin + j * C8NUM, dst_data[i * oc_block + j]); in DepthwiseSWWxKKernel()
1199 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW3x32Kernel() argument
1204 oc_algin *= sizeof(float); in DepthwiseSW3x32Kernel()
1338 : "a"(act_flag), "r"(oc_algin), "r"(dst) in DepthwiseSW3x32Kernel()
1344 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x32Kernel() argument
1348 oc_algin *= sizeof(float); in DepthwiseSW1x32Kernel()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_slidewindow_x86_fp32.cc | 48 int oc_algin = UP_DIV(conv_param_->output_channel_, oc_tile_); in InitPackedInputOutput() local
50 …conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * oc_tile_ * oc_algin; in InitPackedInputOutput()
68 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in Init() local
69 int pack_weight_size = oc_algin * oc_tile_ * weight_tensor->Height() * weight_tensor->Width(); in Init()
169 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in PackWeight() local
172 …NHWCToNXHWCXFp32(weight_tensor->Height(), weight_tensor->Width(), weight_tensor->Batch(), oc_algin, in PackWeight()
179 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in MallocWeightBiasData() local
180 int pack_weight_size = oc_algin * oc_tile_ * weight_tensor->Height() * weight_tensor->Width(); in MallocWeightBiasData()
190 auto bias_size = oc_algin * oc_tile_; in MallocWeightBiasData()
|
| D | convolution_fp32.cc | 64 int oc_algin = UP_DIV(conv_param_->output_channel_, OC_BLOCK); in InitTmpBuffer() local
66 …conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * OC_BLOCK * oc_algin; in InitTmpBuffer()
|