Searched refs:max_batch_size_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | union_find.cc | 57 : batch_size_(absl::nullopt), max_batch_size_(absl::nullopt) {} in ClusterBatchSize()
61 max_batch_size_ == other.max_batch_size_; in operator ==()
92 max_batch_size_ = MergeCompatible<int>(max_batch_size_, max_batch_size); in SetMaxBatchSize()
97 return max_batch_size_; in GetOptionalMaxBatchSize()
102 !CheckIfCompatible(max_batch_size_, other.max_batch_size_)) { in MergeIfCompatible()
107 SetMaxBatchSize(other.max_batch_size_); in MergeIfCompatible()
123 append_optional_num(max_batch_size_); in ToString()
|
| D | union_find.h | 114 absl::optional<int> max_batch_size_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 194 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_)); in BatchFunctionKernel()
290 adaptive_shared_batch_scheduler_options, max_batch_size_, in ComputeAsync()
300 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
409 (i == allowed_batch_sizes_.size() - 1) && (size != max_batch_size_)) { in ValidateAllowedBatchSizes()
425 int32 max_batch_size_; member in tensorflow::BatchFunctionKernel
471 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_)); in BatchKernel()
485 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
515 if (i == allowed_batch_sizes_.size() - 1 && size != max_batch_size_) { in ValidateAllowedBatchSizes()
529 int32 max_batch_size_; member in tensorflow::BatchKernel
|