Searched refs:per_batch_scores (Results 1 – 1 of 1) sorted by relevance
254 Tensor per_batch_scores = inp_scores.Slice(batch, batch + 1); in BatchedNonMaxSuppressionOp() local265 float* scores_data = per_batch_scores.unaligned_flat<float>().data(); in BatchedNonMaxSuppressionOp()