| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | space_to_depth_base.c | 22 int block_size, int h_start, int h_end, int data_size) { in SpaceToDepthForNHWC() argument
29 if (h_start < 0 || h_start >= h_end || h_end > out_shape[1]) { in SpaceToDepthForNHWC()
43 for (int j = h_start; j < h_end; ++j) { in SpaceToDepthForNHWC()
|
| D | batch_to_space_base.c | 60 int h_end = MSMIN((in_h * block_h - crops[1]) / block_h + 1, in_h); in BatchToSpaceForNHWC() local
73 for (int h = h_start; h < h_end; ++h) { in BatchToSpaceForNHWC()
|
| D | space_to_depth_base.h | 25 int block_size, int h_start, int h_end, int data_size);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | resize_fp32.h | 42 const int h_begin, const int h_end);
46 float *line_buffer, const int h_begin, const int h_end);
56 const int h_begin, const int h_end);
|
| D | resize_fp32.c | 235 const int h_end) { in Bilinear() argument
247 for (int h = h_begin; h < h_end; h++) { in Bilinear()
288 const int h_begin, const int h_end) { in ResizeBilinear() argument
305 x_left_weights, line0, line1, h_begin, h_end); in ResizeBilinear()
417 const int h_begin, const int h_end) { in Bicubic() argument
423 for (int h = h_begin; h < h_end; h++) { in Bicubic()
434 float *line_buffer, const int h_begin, const int h_end) { in ResizeBicubic() argument
445 h_end); in ResizeBicubic()
452 const int *y_tops, const int h_begin, const int h_end) { in RewriteExtrapolationValue() argument
473 for (int h = h_begin; h < h_end; ++h) { in RewriteExtrapolationValue()
[all …]
|
| D | conv_depthwise_fp32.c | 42 int h_end = MSMIN(h_start + h_step, conv_param->output_h_); in ConvDw() local
48 for (int oh = h_start; oh < h_end; oh++) { in ConvDw()
845 int h_end = MSMIN(h_start + h_step, conv_param->output_h_); in ConvDwIndirection() local
850 for (int oh = h_start; oh < h_end; oh++) { in ConvDwIndirection()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | crop_and_resize_fp32.cc | 135 int h_end = MSMIN(h_begin + unit, new_height_); in RunImpl() local
136 if (h_end <= h_begin) { in RunImpl()
145 … y_bottom_weights_, x_left_weights_, line0, line1, h_begin, h_end); in RunImpl()
|
| D | resize_fp32.cc | 186 int h_end = std::min(h_begin + unit, new_height_); in RunImpl() local
194 y_weights_, x_weights_, line0, line1, h_begin, h_end); in RunImpl()
204 h_end); in RunImpl()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | space_to_depth_fp32_tests.cc | 42 int h_end = 2; in TEST_F() local
43 …SpaceToDepthForNHWC((const float *)input, output, in_shape, out_shape, 4, 2, h_start, h_end, sizeo… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | batch_to_space_int8.c | 65 int h_end = MSMIN((in_h * block_h - crops[1]) / block_h + 1, in_h); in BatchToSpaceForNHWCInt8() local
83 for (int h = h_start; h < h_end; ++h) { in BatchToSpaceForNHWCInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 403 int h_end = MSMIN(h_start + h_step, conv_param->output_h_); in ConvDwFp16() local
409 for (int oh = h_start; oh < h_end; oh++) { in ConvDwFp16()
|