Searched defs:in_batch_offset (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | pack_tests.cc | 87 int in_batch_offset = b * in_channel * in_h * in_w; in TEST_F() local
150 int in_batch_offset = b * in_channel * in_h * in_w; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 139 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC8Fp16()
183 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC4Fp16()
229 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC1Fp16()
274 int in_batch_offset = batch * in_h * in_w * channel; in MaxPoolingFp16() local
|
| D | conv_fp16.c | 220 int in_batch_offset = b * in_channel * conv_param->input_h_ * conv_param->input_w_; in ConvWinogardFp16() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 44 int in_batch_offset = batch * in_h * in_w * channel; in AvgPoolingInt8() local
99 int in_batch_offset = batch * pooling_param->input_h_ * in_w * channel; in AvgPoolingOptInt8() local
299 int in_batch_offset = batch * in_h * in_w * channel; in MaxPoolingInt8() local
346 int in_batch_offset = batch * in_h * in_w * channel; in MaxPoolingWithQuantInt8() local
434 int in_batch_offset = batch * pooling_param->input_h_ * in_w * channel; in MaxPoolingOptInt8() local
|
| D | conv_int8.c | 54 int in_batch_offset = b * in_channel * conv_param->input_h_ * conv_param->input_w_; in ConvInt8() local
|
| D | conv3x3_int8.c | 885 int in_batch_offset = batch * ic8 * C8NUM * conv_param->input_h_ * conv_param->input_w_; in Conv3x3Int8() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_winograd_fp32.c | 59 int in_batch_offset = b * in_channel * conv_param->input_h_ * conv_param->input_w_; in ConvWinogardFp32() local
|
| D | adder_fp32.c | 69 …int in_batch_offset = b * conv_param->input_channel_ * conv_param->input_h_ * conv_param->input_w_; in AdderFp32() local
|