Searched refs:weight_block (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_fp16.c | 83 int weight_block = UP_DIV(conv_param->output_channel_, C8NUM); in ConvOutNc8hw8Fp16() local
102 for (int j = 0; j < weight_block; j++, cur_weight += C8NUM * deep, cur_bias += C8NUM) { in ConvOutNc8hw8Fp16()
103 … int real_weight_row = (j != weight_block - 1) ? C8NUM : conv_param->output_channel_ - j * C8NUM; in ConvOutNc8hw8Fp16()
123 int weight_block = UP_DIV(param->col_, C8NUM); in Conv1x1OutNc8hw8MultiThreadByInputFp16() local
144 for (int j = 0; j < weight_block; j++, cur_weight += C8NUM * param->deep_, cur_bias += C8NUM) { in Conv1x1OutNc8hw8MultiThreadByInputFp16()
145 int real_weight_row = (j != weight_block - 1) ? C8NUM : param->col_ - j * C8NUM; in Conv1x1OutNc8hw8MultiThreadByInputFp16()
165 int weight_block = UP_DIV(param->col_, C8NUM); in Conv1x1OutNc8hw8MultiThreadByWeightFp16() local
167 int block_per_thread = UP_DIV(weight_block, param->op_parameter_.thread_num_); in Conv1x1OutNc8hw8MultiThreadByWeightFp16()
169 int weight_end_block = MSMIN(weight_start_block + block_per_thread, weight_block); in Conv1x1OutNc8hw8MultiThreadByWeightFp16()
178 int real_weight_row = (j != weight_block - 1) ? C8NUM : param->col_ - j * C8NUM; in Conv1x1OutNc8hw8MultiThreadByWeightFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.h | 39 int weight_block);
42 int weight_block);
|
| D | conv_depthwise_fp32.c | 142 int weight_block) { in InitSlidingParamConv() argument
143 InitSlidingParam(sliding, conv_param, weight_block); in InitSlidingParamConv()
144 AppendSlidingParamConv(sliding, conv_param, input_block, weight_block); in InitSlidingParamConv()
148 int weight_block) { in AppendSlidingParamConv() argument
160 …->kernel_step_ = conv_param->kernel_w_ * conv_param->kernel_h_ * sliding->ic_align_ * weight_block; in AppendSlidingParamConv()
|