Searched refs:scores_batch (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | bounding_box.cu | 588 Tensor<gpu, 1, DType> scores_batch(scores->dptr_ + i * num_elements_per_batch, in CompactNMSResults() local
600 mxnet::op::SortByKey(scores_batch, indices_batch, false, scratch, in CompactNMSResults()
666 Tensor<gpu, 1, DType> scores_batch(scores.dptr_ + i * num_elem, in BoxNMSForwardGPU_notemp()
678 mxnet::op::SortByKey(scores_batch, indices_batch, false, &scratch, 0, in BoxNMSForwardGPU_notemp()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | bounding_box.cu | 588 Tensor<gpu, 1, DType> scores_batch(scores->dptr_ + i * num_elements_per_batch, in CompactNMSResults() local
600 mxnet::op::SortByKey(scores_batch, indices_batch, false, scratch, in CompactNMSResults()
666 Tensor<gpu, 1, DType> scores_batch(scores.dptr_ + i * num_elem, in BoxNMSForwardGPU_notemp()
678 mxnet::op::SortByKey(scores_batch, indices_batch, false, &scratch, 0, in BoxNMSForwardGPU_notemp()
|