Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc287 for (int task_idx = 0; task_idx < batch.num_tasks(); ++task_idx) { in ValidateBatch() local
288 const BatchResourceBase::BatchTask& task = batch.task(task_idx); in ValidateBatch()
345 for (int task_idx = 0; task_idx < batch.num_tasks(); ++task_idx) { in ConcatInputTensors() local
346 to_concatenate.push_back(batch.task(task_idx).inputs.at(i)); in ConcatInputTensors()
663 for (int task_idx = 0; task_idx < batch->num_tasks() - 1; ++task_idx) { in ProcessBatch() local
664 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
674 for (int task_idx = 0; task_idx < batch->num_tasks() - 1; ++task_idx) { in ProcessBatch() local
675 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
684 for (int task_idx = 0; task_idx < batch->num_tasks(); ++task_idx) { in ProcessBatch() local
685 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
[all …]