Home
last modified time | relevance | path

Searched defs:enable_large_batch_splitting (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dbatch_ops.py35 enable_large_batch_splitting=True): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbasic_batch_scheduler.h225 bool enable_large_batch_splitting = false; member
H A Dbatch_resource_base.cc334 bool enable_large_batch_splitting) { in GetBatcherQueueOptions()
364 int32_t max_enqueued_batches, bool enable_large_batch_splitting, in GetAdaptiveBatcherQueueOptions()
H A Dshared_batch_scheduler.h183 bool enable_large_batch_splitting = false; member
H A Dshared_batch_scheduler_test.cc120 bool enable_large_batch_splitting, in CreateQueueOptions()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc120 bool enable_large_batch_splitting, in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc140 bool enable_large_batch_splitting, in Create()