Searched refs:ClusterBatchSize (Results 1 – 3 of 3) sorted by relevance
56 ClusterBatchSize::ClusterBatchSize() in ClusterBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize59 bool ClusterBatchSize::operator==(const ClusterBatchSize& other) { in operator ==()64 ClusterBatchSize& ClusterBatchSize::SetBatchSize(int batch_size) { in SetBatchSize()69 ClusterBatchSize& ClusterBatchSize::SetBatchSize( in SetBatchSize()78 bool ClusterBatchSize::HasBatchSize() const { return batch_size_.has_value(); } in HasBatchSize()80 int ClusterBatchSize::GetBatchSize() const { in GetBatchSize()85 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize(int max_batch_size) { in SetMaxBatchSize()90 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize( in SetMaxBatchSize()96 absl::optional<int> ClusterBatchSize::GetOptionalMaxBatchSize() const { in GetOptionalMaxBatchSize()100 bool ClusterBatchSize::MergeIfCompatible(const ClusterBatchSize& other) { in MergeIfCompatible()[all …]
72 class ClusterBatchSize {74 ClusterBatchSize();76 bool operator==(const ClusterBatchSize& other);77 bool operator!=(const ClusterBatchSize& other) { return !(*this == other); }82 ClusterBatchSize& SetBatchSize(int batch_size);88 ClusterBatchSize& SetMaxBatchSize(int max_batch_size);94 bool MergeIfCompatible(const ClusterBatchSize& other);110 ClusterBatchSize& SetBatchSize(const absl::optional<int>& batch_size);111 ClusterBatchSize& SetMaxBatchSize(const absl::optional<int>& batch_size);118 const ClusterBatchSize& batch_size) {[all …]
634 ClusterBatchSize GetClusterBatchSizeForNode( in GetClusterBatchSizeForNode()637 ClusterBatchSize cluster_batch_size; in GetClusterBatchSizeForNode()825 ClusterBatchSize expected_batch_size = in SegmentGraph()849 ClusterBatchSize out_batch_size = out_cluster->Property().BatchSize(); in SegmentGraph()850 ClusterBatchSize merged_batch_size = expected_batch_size; in SegmentGraph()