Searched defs:data_buff (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | nms_with_mask_gpu_kernel.cc | 77 T *data_buff = GetDeviceAddress<T>(workspace, kIndex0); in LaunchKernel() local
|
| /third_party/ltp/testcases/kernel/syscalls/statx/ |
| D | statx01.c | 201 char data_buff[SIZE]; in setup() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | nmswithmask_tensorrt.cc | 94 void *data_buff = nullptr; in RunCudaNMSwithmask() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | nms_with_mask_impl.cu | 165 T *data_buff, int *index_buff, int box_size) { in NmsBitonicSortByKeyKernel()
222 cudaError_t CalSort(const int &num, const T *data_in, T *data_out, int *index_buff, T *data_buff, i… in CalSort()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | nms_with_mask_cpu_kernel.cc | 93 … T *data_buff, int *index_buff, int box_size) { in NmsBitonicSortByKeyKernel()
249 auto data_buff = reinterpret_cast<T *>(workspace[kIndexDataBuff]->device_ptr()); in LaunchKernel() local
|