Searched refs:nmsed_classes_t (Results 1 – 1 of 1) sorted by relevance
412 Tensor* nmsed_classes_t = nullptr; in BatchedNonMaxSuppressionOp() local414 context->allocate_output(2, scores_shape, &nmsed_classes_t)); in BatchedNonMaxSuppressionOp()415 auto nmsed_classes_flat = nmsed_classes_t->template flat<float>(); in BatchedNonMaxSuppressionOp()