Searched defs:MaskInit (Results 1 – 2 of 2) sorted by relevance
40 __global__ void MaskInit(int numSq, bool *row_mask) { in MaskInit() function
128 void NMSWithMaskCpuKernelMod::MaskInit(size_t numSq, bool *row_mask) { in MaskInit() function in mindspore::kernel::NMSWithMaskCpuKernelMod