Searched refs:SetBatchSize (Results 1 – 3 of 3) sorted by relevance
64 ClusterBatchSize& ClusterBatchSize::SetBatchSize(int batch_size) { in SetBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize65 SetBatchSize(static_cast<absl::optional<int>>(batch_size)); in SetBatchSize()69 ClusterBatchSize& ClusterBatchSize::SetBatchSize( in SetBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize86 SetBatchSize(static_cast<absl::optional<int>>(max_batch_size)); in SetMaxBatchSize()106 SetBatchSize(other.batch_size_); in MergeIfCompatible()
82 ClusterBatchSize& SetBatchSize(int batch_size);110 ClusterBatchSize& SetBatchSize(const absl::optional<int>& batch_size);
664 return cluster_batch_size.SetBatchSize(leading_shape->dim(0).size()); in GetClusterBatchSizeForNode()