Searched defs:SetMaxBatchSize (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ | ||
| D | union_find.cc | 85 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize(int max_batch_size) { in SetMaxBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize |