Home
last modified time | relevance | path

Searched refs:in_plane (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dgroup_convolution_fp32.cc26 int in_plane = in_tensor->Height() * in_tensor->Width() * in_tensor->Batch(); in SeparateInput() local
27 if (in_plane < 0) { in SeparateInput()
37 for (int i = 0; i < in_plane; ++i) { in SeparateInput()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dgroup_convolution_fp16.cc27 int in_plane = in_tensor->Height() * in_tensor->Width() * in_tensor->Batch(); in SeparateInput() local
47 for (int i = 0; i < in_plane; ++i) { in SeparateInput()
57 for (int i = 0; i < in_plane; ++i) { in SeparateInput()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dgroup_convolution_int8.cc24 int in_plane = conv_param_->input_h_ * conv_param_->input_w_ * conv_param_->input_batch_; in SeparateInput() local
31 for (int i = 0; i < in_plane; ++i) { in SeparateInput()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dresize_int8.c27 int in_plane = in_h * in_w; in ResizeBilinearInt8() local
30 const int8_t *in_b_ptr = input_ptr + n * in_plane * channel; in ResizeBilinearInt8()
76 int in_plane = in_h * in_w; in ResizeBilinearWithFloatScaleInt8() local
79 const int8_t *in_b_ptr = input_ptr + n * in_plane * channel; in ResizeBilinearWithFloatScaleInt8()
Dpack_int8.c819 int in_plane = in_h * in_w; in PackInputToC8Int8() local
822 int src_batch_offset = b * in_channel * in_plane; in PackInputToC8Int8()
823 int dst_batch_offset = b * ic8_round * in_plane; in PackInputToC8Int8()
824 for (int k = 0; k < in_plane; k++) { in PackInputToC8Int8()
829 int dst_c_offset = dst_plane_offset + i * in_plane; in PackInputToC8Int8()
839 int tmp_ic_offset = ic8 * in_plane; in PackInputToC8Int8()