Home
last modified time | relevance | path

Searched refs:c_block_ (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h65 int c_block_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_1x1_x86_fp32.c54 int oc_num = sw_param->c_block_; in Conv1x1SWFp32()
Dconv_depthwise_fp32.c123 sliding->c_block_ = UP_DIV(conv_param->output_channel_, block); in InitSlidingParam()
315 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in ConvDwSWFp32()
985 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in DeconvDwSWFp32()
1070 int oc_num = sw_param->c_block_; in DepthwiseSWAvxFp32()
Dconv_common_fp32.c365 int c_block = sw_param->c_block_; in ConvSWFp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.c581 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in ConvDwC8Fp16()
741 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in DeconvDwC8Fp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.c627 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in ConvDwInt8SW()
780 for (int oc = task_id; oc < sliding->c_block_; oc += conv_param->thread_num_) { in DeconvDwInt8()