Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc432 const int output_task_num = output_task_sizes.size(); in SplitInputTask() local
433 input_task.output->resize(output_task_num); in SplitInputTask()
435 for (int i = 0; i < output_task_num; ++i) { in SplitInputTask()
439 output_tasks->reserve(output_task_num); in SplitInputTask()
440 for (int i = 0; i < output_task_num; i++) { in SplitInputTask()