Searched refs:nmsed_scores_flat (Results 1 – 1 of 1) sorted by relevance
410 auto nmsed_scores_flat = nmsed_scores_t->template flat<float>(); in BatchedNonMaxSuppressionOp() local426 nmsed_scores_flat(i * per_batch_size + j) = nmsed_scores[i][j]; in BatchedNonMaxSuppressionOp()