Searched refs:valid_detections_shape (Results 1 – 2 of 2) sorted by relevance
71 valid_detections_shape = [batch_size]75 nmsed_classes_shape, valid_detections_shape187 valid_detections_shape = [batch_size]199 nmsed_classes_shape, valid_detections_shape
527 TensorShape valid_detections_shape({num_batches}); in BatchedNonMaxSuppressionOp() local528 OP_REQUIRES_OK(context, context->allocate_output(3, valid_detections_shape, in BatchedNonMaxSuppressionOp()