Searched refs:mutable_task (Results 1 – 5 of 5) sorted by relevance
111 TaskType* mutable_task(int i);291 TaskType* Batch<TaskType>::mutable_task(int i) { in mutable_task() function
608 BatchTask& task = *(batch->mutable_task(j)); in SplitOutputTensors()655 batch->mutable_task(i)->status->Update(status); in ProcessFuncBatch()657 batch->mutable_task(i)->context->SetStatus(status); in ProcessFuncBatch()660 batch->mutable_task(i)->done_callback(); in ProcessFuncBatch()796 batch->mutable_task(task_idx)->done_callback(); in ProcessBatch()
64 EXPECT_EQ(task1->size(), batch.mutable_task(1)->size()); in TEST()
288 batch_data.push_back(batch->mutable_task(i)->size()); in TEST_P()
776 TaskDef* task_def = req.mutable_task(); in AssignTask()