Searched refs:output_tasks (Results 1 – 6 of 6) sorted by relevance
188 std::vector<std::unique_ptr<TaskType>>* output_tasks)>273 std::vector<std::unique_ptr<TaskType>>* output_tasks)>;342 std::vector<std::unique_ptr<TaskType>>* output_tasks)723 std::vector<std::unique_ptr<TaskType>> output_tasks; in ScheduleWithSplit() local727 output_tasks.push_back(std::move(*task)); in ScheduleWithSplit()729 TF_RETURN_IF_ERROR(SplitInputBatchIntoSubtasks(task, &output_tasks)); in ScheduleWithSplit()732 for (int i = 0; i < output_tasks.size(); ++i) { in ScheduleWithSplit()733 if (batches_.back()->size() + output_tasks[i]->size() > in ScheduleWithSplit()741 [&output_tasks, i] { in ScheduleWithSplit()743 {{"size", output_tasks[i]->size()}}); in ScheduleWithSplit()[all …]
241 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetBatcherQueueOptions() argument243 max_batch_size, output_tasks); in GetBatcherQueueOptions()277 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetAdaptiveBatcherQueueOptions() argument279 max_batch_size, output_tasks); in GetAdaptiveBatcherQueueOptions()381 int max_batch_size, std::vector<std::unique_ptr<BatchTask>>* output_tasks) { in SplitInputTask() argument439 output_tasks->reserve(output_task_num); in SplitInputTask()441 output_tasks->push_back(input_task.CreateSplitTask(i, barrier.Inc())); in SplitInputTask()467 for (int j = 0; j < output_tasks->size(); ++j) { in SplitInputTask()468 BatchTask& output_task = *((*output_tasks)[j]); in SplitInputTask()
392 std::vector<std::unique_ptr<FakeTask>>* output_tasks) { in TEST() argument394 output_tasks->push_back(std::move(*input_task)); in TEST()395 int remaining_size = output_tasks->back()->size() - first_size; in TEST()396 output_tasks->back()->set_size(first_size); in TEST()399 output_tasks->emplace_back(new FakeTask(task_size)); in TEST()
245 std::vector<std::unique_ptr<TaskType>>* output_tasks)>
188 std::vector<std::unique_ptr<BatchTask>>* output_tasks);
156 std::vector<std::unique_ptr<TaskType>>* output_tasks)>