Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc301 for (int task_idx = 0; task_idx < batch.num_tasks(); ++task_idx) { in ValidateBatch() local
302 const BatchTask& task = batch.task(task_idx); in ValidateBatch()
348 for (int task_idx = 0; task_idx < batch.num_tasks(); ++task_idx) { in ConcatInputTensors() local
349 to_concatenate.push_back(batch.task(task_idx).inputs.at(i)); in ConcatInputTensors()
581 for (int task_idx = 0; task_idx < batch->num_tasks() - 1; ++task_idx) { in ProcessBatch() local
582 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
593 for (int task_idx = 0; task_idx < batch->num_tasks() - 1; ++task_idx) { in ProcessBatch() local
594 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
603 for (int task_idx = 0; task_idx < batch->num_tasks(); ++task_idx) { in ProcessBatch() local
604 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
[all …]