Searched refs:nmsed_boxes_t (Results 1 – 1 of 1) sorted by relevance
400 Tensor* nmsed_boxes_t = nullptr; in BatchedNonMaxSuppressionOp() local403 context->allocate_output(0, boxes_shape, &nmsed_boxes_t)); in BatchedNonMaxSuppressionOp()404 auto nmsed_boxes_flat = nmsed_boxes_t->template flat<float>(); in BatchedNonMaxSuppressionOp()