Searched refs:d_nms_mask (Results 1 – 1 of 1) sorted by relevance
262 Tensor h_num_selected, d_nms_mask; in NmsGpu() local268 DataType::DT_INT32, TensorShape({max_nms_mask_size}), &d_nms_mask)); in NmsGpu()271 auto config = GetGpuLaunchConfig(d_nms_mask.NumElements(), device); in NmsGpu()275 d_nms_mask.flat<int32>().data())); in NmsGpu()285 int* d_delete_mask = d_nms_mask.flat<int>().data(); in NmsGpu()