Home
last modified time | relevance | path

Searched defs:NmsBitonicSortByKeyKernel (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dnms_with_mask_impl.cu164 __global__ void NmsBitonicSortByKeyKernel(const int outer, const int inner, const int ceil_power2, … in NmsBitonicSortByKeyKernel() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dnms_with_mask_cpu_kernel.cc92 void NMSWithMaskCpuKernelMod::NmsBitonicSortByKeyKernel(const int inner, const size_t ceil_power2, … in NmsBitonicSortByKeyKernel() function in mindspore::kernel::NMSWithMaskCpuKernelMod