Home
last modified time | relevance | path

Searched defs:score_threshold_ (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dnon_max_suppression.h29 float score_threshold_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcombined_non_max_suppression_cpu_kernel.h88 float score_threshold_ = 0.0; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dnon_max_suppression_fp32.h49 float score_threshold_ = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dnon_max_suppression_fp32_tests.cc41 float score_threshold_; member in mindspore::TestNMSFp32
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dnonmaxsuppressionv3_helper.h82 M *score_threshold_ = nullptr; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dnon_max_suppressionv3_impl.cu234 … M score_threshold_, int *index_buff, S max_output_size_, int box_size, unsigned int *sel_mask, in DoNms()