Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler.h253 split_input_task_func; member
333 shared_scheduler_queue_options.split_input_task_func = in Create()
334 options.split_input_task_func; in Create()
Dadaptive_shared_batch_scheduler.h168 split_input_task_func; member
703 if (options_.split_input_task_func == nullptr && in Schedule()
729 if (options_.split_input_task_func == nullptr || in Schedule()
738 TF_RETURN_IF_ERROR(options_.split_input_task_func( in Schedule()
Dshared_batch_scheduler.h193 split_input_task_func; member
557 options.split_input_task_func == nullptr) { in AddQueue()
759 options_.split_input_task_func); in ScheduleWithLazySplit()
1085 return options_.split_input_task_func( in SplitInputBatchIntoSubtasks()
Dbatch_resource_base.cc305 batcher_queue_options.split_input_task_func = in GetBatcherQueueOptions()
341 batcher_queue_options.split_input_task_func = in GetAdaptiveBatcherQueueOptions()
Dadaptive_shared_batch_scheduler_test.cc460 queue_options.split_input_task_func = in TEST()
Dshared_batch_scheduler_test.cc129 queue_options.split_input_task_func = split_func; in CreateQueueOptions()