Searched refs:end_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 | 69 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute() local
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 | 103 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute() local
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
376 for (oh = start_oh + 1; oh < end_oh - 1; oh++) { in ConvDw3x3Fp16()
|
| /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
700 for (oh = start_oh + 1; oh < end_oh - 1; oh++) { in ConvDw3x3()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 303 int end_oh = MSMIN(start_oh + step_oh, sliding->bottom_); in ConvDw3x3Int8() local
322 for (int oh = start_oh; oh < end_oh; oh++) { in ConvDw3x3Int8()
|