Home
last modified time | relevance | path

Searched refs:in_channel_offset (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c53 int in_channel_offset = in_batch_offset + j; in AvgPoolingInt8() local
62 … int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPoolingInt8()
134 int in_channel_offset = in_batch_offset + j * C16NUM; in AvgPoolingOptInt8() local
139 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
207 int in_channel_offset = in_c16_offset + j * C8NUM; in AvgPoolingOptInt8() local
211 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
256 int in_channel_offset = in_c8_offset + k; in AvgPoolingOptInt8() local
261 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
308 int in_channel_offset = in_batch_offset + j; in MaxPoolingInt8() local
316 … int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingInt8()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c149 int in_channel_offset = in_batch_offset + j * C8NUM; in MaxPoolingC8Fp16() local
158 int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingC8Fp16()
196 int in_channel_offset = in_batch_offset + c4_offset + j * C4NUM; in MaxPoolingC4Fp16() local
205 int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingC4Fp16()
238 int in_channel_offset = in_batch_offset + k; in MaxPoolingC1Fp16() local
243 int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingC1Fp16()