Searched refs:dev_sorted_scores (Results 1 – 1 of 1) sorted by relevance
370 Tensor dev_sorted_scores; // sorted scores, cub output in Compute() local381 &d_sorted_conv_layer_indexes, &dev_sorted_scores, &dev_boxes, in Compute()400 scores.flat<float>().data(), dev_sorted_scores.flat<float>().data(), in Compute()453 float* d_sorted_scores = dev_sorted_scores.flat<float>().data(); in Compute()