Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c90 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPoolingOptInt8() local
91 …int thread_num = out_tile_count < pooling_param->thread_num_ ? out_tile_count : pooling_param->thr… in AvgPoolingOptInt8()
101 for (int thread_id = task_id; thread_id < out_tile_count; thread_id += thread_num) { in AvgPoolingOptInt8()
334 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPoolingWithQuantInt8() local
335 …int thread_num = out_tile_count < pooling_param->thread_num_ ? out_tile_count : pooling_param->thr… in MaxPoolingWithQuantInt8()
348 for (int thread_id = task_id; thread_id < out_tile_count; thread_id += thread_num) { in MaxPoolingWithQuantInt8()
428 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPoolingOptInt8() local
429 int thread_num = MSMIN(out_tile_count, pooling_param->thread_num_); in MaxPoolingOptInt8()
436 for (int thread_id = task_id; thread_id < out_tile_count; thread_id += thread_num) { in MaxPoolingOptInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c32 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPooling() local
48 …for (int thread_id = task_id; thread_id < out_tile_count; thread_id += pooling_param->thread_num_)… in AvgPooling()
147 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPooling() local
163 …for (int thread_id = task_id; thread_id < out_tile_count; thread_id += pooling_param->thread_num_)… in MaxPooling()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c31 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPoolingFp16() local
42 …for (int thread_id = task_id; thread_id < out_tile_count; thread_id += pooling_param->thread_num_)… in AvgPoolingFp16()
268 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPoolingFp16() local
276 for (int thread_id = task_id; thread_id < out_tile_count; thread_id += thread_num) { in MaxPoolingFp16()