Home
last modified time | relevance | path

Searched refs:split_input_task_func (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbasic_batch_scheduler.h252 split_input_task_func; member
332 shared_scheduler_queue_options.split_input_task_func = in Create()
333 options.split_input_task_func; in Create()
H A Dadaptive_shared_batch_scheduler.h170 split_input_task_func; member
725 if (options_.split_input_task_func == nullptr && in Schedule()
751 if (options_.split_input_task_func == nullptr || in Schedule()
760 TF_RETURN_IF_ERROR(options_.split_input_task_func( in Schedule()
H A Dshared_batch_scheduler.h200 split_input_task_func; member
593 options.split_input_task_func == nullptr) { in AddQueueAfterRewritingOptions()
832 options_.split_input_task_func); in ScheduleWithLazySplit()
1146 return options_.split_input_task_func( in SplitInputBatchIntoSubtasks()
H A Dbatch_resource_base.cc342 batcher_queue_options.split_input_task_func = in GetBatcherQueueOptions()
378 batcher_queue_options.split_input_task_func = in GetAdaptiveBatcherQueueOptions()
H A Dadaptive_shared_batch_scheduler_test.cc460 queue_options.split_input_task_func = in TEST()
H A Dshared_batch_scheduler_test.cc130 queue_options.split_input_task_func = split_func; in CreateQueueOptions()