Searched refs:d_num_selected (Results 1 – 1 of 1) sorted by relevance
341 Tensor d_num_selected; in NmsGpu() local343 TensorShape({1}), &d_num_selected)); in NmsGpu()351 d_num_selected.flat<int>().data(), num_boxes, device.stream()); in NmsGpu()355 device.memcpyDeviceToHost(h_selected_count, d_num_selected.flat<int>().data(), in NmsGpu()