Searched refs:leading_shape (Results 1 – 1 of 1) sorted by relevance
531 absl::optional<const TensorShapeProto*> leading_shape = in OperationCanBeTranslatedToImplicitBatch() local533 return leading_shape.has_value() && leading_shape.value()->dim_size() >= 2; in OperationCanBeTranslatedToImplicitBatch()661 const TensorShapeProto* leading_shape = optional_leading_shape.value(); in GetClusterBatchSizeForNode() local662 DCHECK(!leading_shape->unknown_rank() && leading_shape->dim_size() >= 2); in GetClusterBatchSizeForNode()663 VLOG(3) << "set batch size as " << leading_shape->dim(0).size(); in GetClusterBatchSizeForNode()664 return cluster_batch_size.SetBatchSize(leading_shape->dim(0).size()); in GetClusterBatchSizeForNode()