Searched refs:nmsed_scores_t (Results 1 – 1 of 1) sorted by relevance
406 Tensor* nmsed_scores_t = nullptr; in BatchedNonMaxSuppressionOp() local409 context->allocate_output(1, scores_shape, &nmsed_scores_t)); in BatchedNonMaxSuppressionOp()410 auto nmsed_scores_flat = nmsed_scores_t->template flat<float>(); in BatchedNonMaxSuppressionOp()