Home
last modified time | relevance | path

Searched refs:loop_num_ (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dtopk_int8.cc41 parameter->loop_num_ = 1; in ReSize()
43 parameter->loop_num_ *= input->shape().at(i); in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dtopk_fp32.cc40 topk_param_->loop_num_ = 1; in ReSize()
42 topk_param_->loop_num_ *= input->shape().at(i); in ReSize()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dtopk_fp32.h35 int loop_num_; member
Dtopk_fp32.c43 int loop_num = parameter->loop_num_; in Topk()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dtopk_int8.c29 int loop_num = parameter->loop_num_; in TopkInt8()