Searched defs:sel_idx (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | nms_with_mask_gpu_kernel.h | 57 int *sel_idx = GetDeviceAddress<int>(outputs, 1); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | nms_with_mask_impl.cu | 100 __global__ void Preprocess(const int num, int *sel_idx, bool *sel_boxes, T *output, int box_size) { in Preprocess()
174 void CalPreprocess(const int num, int *sel_idx, bool *sel_boxes, T *input, T *output, int *index_bu… in CalPreprocess()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | nms_with_mask_cpu_kernel.cc | 125 void NMSWithMaskCPUKernel<T>::Preprocess(const int num, int *sel_idx, bool *sel_boxes) { in Preprocess()
230 auto sel_idx = reinterpret_cast<int *>(outputs[SEL_IDX]->addr); in Launch() local
|