Home
last modified time | relevance | path

Searched refs:full_batch_scheduling_boost_micros (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dserial_device_batch_scheduler.h83 int64_t full_batch_scheduling_boost_micros = 0; member
281 if (options.full_batch_scheduling_boost_micros < 0) { in Create()
284 options.full_batch_scheduling_boost_micros); in Create()
390 options_.full_batch_scheduling_boost_micros * (*best_it)->size() / in ProcessBatches()
395 options_.full_batch_scheduling_boost_micros * (*it)->size() / in ProcessBatches()
H A Dadaptive_shared_batch_scheduler.h116 int64_t full_batch_scheduling_boost_micros = 0; member
395 if (options.full_batch_scheduling_boost_micros < 0) { in Create()
398 options.full_batch_scheduling_boost_micros); in Create()
569 options_.full_batch_scheduling_boost_micros * (*it)->size() / in MaybeScheduleNextBatch()
H A Dserial_device_batch_scheduler_test.cc206 options.full_batch_scheduling_boost_micros = 10; in TEST()
351 options.full_batch_scheduling_boost_micros = 1000; in TEST()
H A Dadaptive_shared_batch_scheduler_test.cc213 options.full_batch_scheduling_boost_micros = 100; in TEST()
287 options.full_batch_scheduling_boost_micros = 0; in TEST()