Home
last modified time | relevance | path

Searched refs:CreateSplitTask (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.h100 std::unique_ptr<BatchTask> CreateSplitTask(
H A Dbatch_resource_base.cc211 BatchResourceBase::BatchTask::CreateSplitTask( in CreateSplitTask() function in tensorflow::serving::BatchResourceBase::BatchTask
548 output_tasks->push_back(input_task.CreateSplitTask(i, barrier.Inc())); in SplitInputTask()