Home
last modified time | relevance | path

Searched refs:enable_lazy_split (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dshared_batch_scheduler.h212 bool enable_lazy_split = false; member
600 if (options.enable_lazy_split && (!options.enable_large_batch_splitting)) { in AddQueueAfterRewritingOptions()
772 if (options_.enable_lazy_split) { in Queue()
786 if (options_.enable_lazy_split) { in ~Queue()
800 if (options_.enable_lazy_split) { in Schedule()
954 if (options_.enable_lazy_split) { in NumEnqueuedTasks()
1045 if (!options_.enable_lazy_split) { in ScheduleBatch()
1119 if (options_.enable_lazy_split) { in IsEmptyInternal()
1130 if (options_.enable_lazy_split) { in StartNewBatch()
1164 if (!options_.enable_lazy_split) { in IsOpenBatchSchedulable()
[all …]
H A Dbasic_batch_scheduler.h232 bool enable_lazy_split = false; member
334 shared_scheduler_queue_options.enable_lazy_split = options.enable_lazy_split; in Create()
H A Dshared_batch_scheduler_test.cc121 bool enable_lazy_split, SplitFunc split_func) { in CreateQueueOptions() argument
128 queue_options.enable_lazy_split = enable_lazy_split; in CreateQueueOptions()
144 max_enqueued_batches, enable_input_batch_split(), enable_lazy_split(), in CreateQueueOptions()
149 bool enable_lazy_split() const { return std::get<1>(GetParam()); } in enable_lazy_split() function in tensorflow::serving::__anon3314b7bb0111::SharedBatchSchedulerTest
778 enable_input_batch_split(), enable_lazy_split(), in TEST_P()
788 enable_input_batch_split(), enable_lazy_split(), get_split_func()), in TEST_P()