Home
last modified time | relevance | path

Searched defs:output_tasks (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_input_task.h356 std::vector<std::unique_ptr<TaskType>>* output_tasks) { in SplitBatches()
Dshared_batch_scheduler.h834 std::vector<std::unique_ptr<TaskType>> output_tasks; in ScheduleWithoutOrEagerSplit() local
1082 std::vector<std::unique_ptr<TaskType>>* output_tasks) { in SplitInputBatchIntoSubtasks()
Dbatch_resource_base.cc308 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetBatcherQueueOptions()
344 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetAdaptiveBatcherQueueOptions()
452 int max_batch_size, std::vector<std::unique_ptr<BatchTask>>* output_tasks) { in SplitInputTask()
Dbatch_input_task_test.cc253 output_tasks; in TEST_F() local
Dshared_batch_scheduler_test.cc157 std::vector<std::unique_ptr<FakeTask>>* output_tasks) -> Status { in get_split_func()
842 std::vector<std::unique_ptr<FakeTask>>* output_tasks) -> Status { in CreateQueues()
Dadaptive_shared_batch_scheduler_test.cc462 std::vector<std::unique_ptr<FakeTask>>* output_tasks) { in TEST()