Searched refs:d_indices (Results 1 – 1 of 1) sorted by relevance
315 Tensor d_indices; in NmsGpu() local317 DataType::DT_INT32, TensorShape({num_boxes}), &d_indices)); in NmsGpu()321 d_indices.flat<int>().data())); in NmsGpu()348 d_indices.flat<int>().data(), // input in NmsGpu()443 Tensor d_indices; in DoNMS() local445 DataType::DT_INT32, TensorShape({num_boxes}), &d_indices)); in DoNMS()465 d_indices.flat<int>().data())); in DoNMS()470 d_indices.flat<int>().data(), d_sorted_indices.flat<int>().data(), in DoNMS()