Searched refs:SetMaxBatchSize (Results 1 – 3 of 3) sorted by relevance
73 SetMaxBatchSize(batch_size_); in SetBatchSize()85 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize(int max_batch_size) { in SetMaxBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize90 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize( in SetMaxBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize107 SetMaxBatchSize(other.max_batch_size_); in MergeIfCompatible()
88 ClusterBatchSize& SetMaxBatchSize(int max_batch_size);111 ClusterBatchSize& SetMaxBatchSize(const absl::optional<int>& batch_size);
644 cluster_batch_size.SetMaxBatchSize( in GetClusterBatchSizeForNode()