Home
last modified time | relevance | path

Searched defs:max_enqueued_batches (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dshared_batch_scheduler_test.cc119 size_t max_enqueued_batches, in CreateQueueOptions()
141 size_t max_enqueued_batches) { in CreateQueueOptions()
209 const size_t max_enqueued_batches = 2; in TEST_P() local
275 const size_t max_enqueued_batches = 2; in TEST_P() local
359 const size_t max_enqueued_batches = 2; in TEST_P() local
417 const size_t max_enqueued_batches = 2; in TEST_P() local
462 const size_t max_enqueued_batches = 2; in TEST_P() local
636 const size_t max_enqueued_batches = 2; in TEST_P() local
692 const size_t max_enqueued_batches = 2; in TEST_P() local
744 const size_t max_enqueued_batches = 2; in TEST_P() local
[all …]
H A Dbatch_resource_base.cc181 void RecordBatchParamMaxEnqueuedBatches(int64_t max_enqueued_batches, in RecordBatchParamMaxEnqueuedBatches()
332 int32_t batch_timeout_micros, int32_t max_enqueued_batches, in GetBatcherQueueOptions()
364 int32_t max_enqueued_batches, bool enable_large_batch_splitting, in GetAdaptiveBatcherQueueOptions()
H A Dbasic_batch_scheduler.h204 int max_enqueued_batches = 10; member
H A Dserial_device_batch_scheduler.h110 int max_enqueued_batches = 10; member
H A Dadaptive_shared_batch_scheduler.h155 int max_enqueued_batches = 10; member
H A Dshared_batch_scheduler.h176 size_t max_enqueued_batches = 10; member
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc117 int32_t max_enqueued_batches, in Create()
141 int32_t max_enqueued_batches, ArrayRef<int32_t> allowed_batch_sizes, in Create()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dbatch_ops.py33 max_enqueued_batches=10, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc136 int32_t max_enqueued_batches, in Create()
160 int32_t max_enqueued_batches, in Create()