Searched refs:input_task (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 239 [](std::unique_ptr<BatchTask>* input_task, in GetBatcherQueueOptions()
242 return SplitInputTask(input_task, open_batch_remaining_slot, in GetBatcherQueueOptions()
275 [](std::unique_ptr<BatchTask>* input_task, in GetAdaptiveBatcherQueueOptions()
278 return SplitInputTask(input_task, open_batch_remaining_slot, in GetAdaptiveBatcherQueueOptions()
382 BatchTask& input_task = *(*input_task_ptr); in SplitInputTask() local
383 const int64 input_task_size = input_task.size(); in SplitInputTask()
387 std::shared_ptr<ThreadSafeStatus> shared_status = input_task.status; in SplitInputTask()
392 [done_callback = input_task.done_callback, output = input_task.output, in SplitInputTask()
393 op_kernel_context = input_task.context, status = shared_status]() { in SplitInputTask()
433 input_task.output->resize(output_task_num); in SplitInputTask()
[all …]
|
| D | shared_batch_scheduler.h | 186 std::function<Status(std::unique_ptr<TaskType>* input_task,
271 std::unique_ptr<TaskType>* input_task, int open_batch_remaining_slot,
341 std::unique_ptr<TaskType>* input_task,
869 std::unique_ptr<TaskType>* input_task, in SplitInputBatchIntoSubtasks() argument
874 std::move(input_task), open_batch_remaining_slot, in SplitInputBatchIntoSubtasks()
|
| D | basic_batch_scheduler.h | 243 std::function<Status(std::unique_ptr<TaskType>* input_task,
|
| D | adaptive_shared_batch_scheduler_test.cc | 391 [](std::unique_ptr<FakeTask>* input_task, int first_size, int max_size, in TEST()
394 output_tasks->push_back(std::move(*input_task)); in TEST()
|
| D | adaptive_shared_batch_scheduler.h | 154 std::function<Status(std::unique_ptr<TaskType>* input_task, int first_size,
|
| /external/pigweed/pw_trace/example/ |
| D | sample_app.cc | 174 } input_task; variable
181 std::array<SimpleRunnable*, 2> tasks = {&input_task, &processing_task}; in StartFakeKernel()
|