Home
last modified time | relevance | path

Searched refs:cal_start_index (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c49 int cal_start_index = thread_id * TILE_NUM; in AvgPooling() local
50 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPooling()
52 int index = cal_start_index + i; in AvgPooling()
164 int cal_start_index = thread_id * TILE_NUM; in MaxPooling() local
165 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPooling()
167 int index = cal_start_index + i; in MaxPooling()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c43 int cal_start_index = thread_id * TILE_NUM; in AvgPoolingFp16() local
44 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPoolingFp16()
46 int index = cal_start_index + i; in AvgPoolingFp16()
277 int cal_start_index = thread_id * TILE_NUM; in MaxPoolingFp16() local
278 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPoolingFp16()
280 int index = cal_start_index + i; in MaxPoolingFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c102 int cal_start_index = thread_id * TILE_NUM; in AvgPoolingOptInt8() local
103 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPoolingOptInt8()
105 int index = cal_start_index + i; in AvgPoolingOptInt8()
349 int cal_start_index = thread_id * TILE_NUM; in MaxPoolingWithQuantInt8() local
350 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPoolingWithQuantInt8()
352 int index = cal_start_index + i; in MaxPoolingWithQuantInt8()
437 int cal_start_index = thread_id * TILE_NUM; in MaxPoolingOptInt8() local
438 int real_cal_num = out_plane - cal_start_index; in MaxPoolingOptInt8()
441 int index = cal_start_index + i; in MaxPoolingOptInt8()