Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dtopk_int8.c28 int last_dim_size = parameter->last_dim_size_; in TopkInt8() local
37 for (int j = 0; j < last_dim_size; j++) { in TopkInt8()
42 qsort(top_map, last_dim_size, sizeof(top_map[0]), DescendCmpInt8); in TopkInt8()
44 qsort(top_map, last_dim_size, sizeof(top_map[0]), AscendCmpInt8); in TopkInt8()
50 cur_input_data += last_dim_size; in TopkInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dtopk_fp32.c42 int last_dim_size = parameter->last_dim_size_; in Topk() local
51 for (int j = 0; j < last_dim_size; j++) { in Topk()
55 qsort(top_map, last_dim_size, sizeof(top_map[0]), DescendCmp); in Topk()
63 cur_input_data += last_dim_size; in Topk()