Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler.h102 TaskType* mutable_task(int i);
246 TaskType* Batch<TaskType>::mutable_task(int i) { in mutable_task() function
Dbatch_scheduler_test.cc64 EXPECT_EQ(task1->size(), batch.mutable_task(1)->size()); in TEST()
Dshared_batch_scheduler_test.cc143 batch_data.push_back(batch->mutable_task(i)->size()); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc464 BatchTask& task = *(batch->mutable_task(j)); in SplitOutputTensors()
491 batch->mutable_task(i)->context->SetStatus(status); in ProcessFuncBatch()
492 batch->mutable_task(i)->done_callback(); in ProcessFuncBatch()
620 batch->mutable_task(task_idx)->done_callback(); in ProcessBatch()