Searched refs:batch_input_task (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | BUILD | 32 name = "batch_input_task",
33 hdrs = ["batch_input_task.h"],
51 ":batch_input_task",
133 ":batch_input_task",
147 ":batch_input_task",
|
| H A D | batch_input_task.h | 59 std::shared_ptr<BatchInputTask<TaskType>> batch_input_task, int split_id,
199 std::shared_ptr<BatchInputTask<TaskType>> batch_input_task, int split_id, in BatchInputTaskHandle() argument
201 : batch_input_task_(batch_input_task), in BatchInputTaskHandle()
|
| H A D | batch_input_task_test.cc | 164 auto batch_input_task = in TEST_F() local
172 batch_input_task->ToTaskHandles(&output_tasks); in TEST_F()
|