Searched refs:d_sorted_indices (Results 1 – 1 of 1) sorted by relevance
446 Tensor d_sorted_indices; in DoNMS() local448 DataType::DT_INT32, TensorShape({num_boxes}), &d_sorted_indices)); in DoNMS()470 d_indices.flat<int>().data(), d_sorted_indices.flat<int>().data(), in DoNMS()481 const int* sorted_indices = d_sorted_indices.flat<int>().data(); in DoNMS()