Searched refs:start_oh (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_3x3_fp32.cc | 68 int start_oh = step_oh * task_id; in DoExecute() local
69 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute()
71 reinterpret_cast<float *>(bias_data_), conv_param_, start_oh, end_oh); in DoExecute()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_depthwise_3x3_fp16.cc | 102 int start_oh = step_oh * task_id; in DoExecute() local
103 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute()
105 reinterpret_cast<float16_t *>(bias_data_), conv_param_, start_oh, end_oh); in DoExecute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.h | 58 … const float16_t *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh);
|
| D | conv_depthwise_fp16.c | 350 … const float16_t *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh) { in ConvDw3x3Fp16() argument
365 int oh = start_oh; in ConvDw3x3Fp16()
376 for (oh = start_oh + 1; oh < end_oh - 1; oh++) { in ConvDw3x3Fp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 302 int start_oh = step_oh * task_id + sliding->top_; in ConvDw3x3Int8() local
303 int end_oh = MSMIN(start_oh + step_oh, sliding->bottom_); in ConvDw3x3Int8()
313 int start_ih = start_oh * conv_param->stride_h_ - conv_param->pad_u_; in ConvDw3x3Int8()
319 start_oh * conv_param->output_w_ * conv_param->output_channel_ + in ConvDw3x3Int8()
322 for (int oh = start_oh; oh < end_oh; oh++) { in ConvDw3x3Int8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.h | 126 const float *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh);
|
| D | conv_depthwise_fp32.c | 674 const float *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh) { in ConvDw3x3() argument
689 int oh = start_oh; in ConvDw3x3()
700 for (oh = start_oh + 1; oh < end_oh - 1; oh++) { in ConvDw3x3()
|