Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler.h246 split_input_task_func; member
326 shared_scheduler_queue_options.split_input_task_func = in Create()
327 options.split_input_task_func; in Create()
Dadaptive_shared_batch_scheduler.h157 split_input_task_func; member
622 if (options_.split_input_task_func == nullptr && in Schedule()
648 if (options_.split_input_task_func == nullptr || in Schedule()
657 TF_RETURN_IF_ERROR(options_.split_input_task_func( in Schedule()
Dshared_batch_scheduler.h189 split_input_task_func; member
479 options.split_input_task_func == nullptr) { in AddQueue()
873 return options_.split_input_task_func( in SplitInputBatchIntoSubtasks()
Dbatch_resource_base.cc238 batcher_queue_options.split_input_task_func = in GetBatcherQueueOptions()
274 batcher_queue_options.split_input_task_func = in GetAdaptiveBatcherQueueOptions()
Dadaptive_shared_batch_scheduler_test.cc390 queue_options.split_input_task_func = in TEST()