Searched refs:valid_detections_shape (Results 1 – 1 of 1) sorted by relevance
418 TensorShape valid_detections_shape({num_batches}); in BatchedNonMaxSuppressionOp() local419 OP_REQUIRES_OK(context, context->allocate_output(3, valid_detections_shape, in BatchedNonMaxSuppressionOp()