Searched refs:h_num_selected (Results 1 – 1 of 1) sorted by relevance
262 Tensor h_num_selected, d_nms_mask; in NmsGpu() local283 DataType::DT_INT32, TensorShape({1}), &h_num_selected, alloc_attr)); in NmsGpu()286 int* h_selected_count = h_num_selected.flat<int>().data(); in NmsGpu()