Searched defs:max_batch_size_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ | ||
| D | union_find.h | 114 absl::optional<int> max_batch_size_; variable |
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | batch_kernels.cc | 425 int32 max_batch_size_; member in tensorflow::BatchFunctionKernel |