Home
last modified time | relevance | path

Searched defs:PopulateOutput (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddense_to_dense_set_operation_cpu_kernel.cc195 bool DenseToDenseSetOperationCpuKernelMod::PopulateOutput(const std::vector<kernel::KernelTensor *>… in PopulateOutput() function in mindspore::kernel::DenseToDenseSetOperationCpuKernelMod
Dnms_with_mask_cpu_kernel.cc140 void NMSWithMaskCpuKernelMod::PopulateOutput(const T *data_in, T *data_out, const int *index_buff, … in PopulateOutput() function in mindspore::kernel::NMSWithMaskCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dnms_with_mask_impl.cu49 __global__ void PopulateOutput(const T *data_in, T *data_out, int *index_buff, const int num, int b… in PopulateOutput() function