Searched refs:mutable_task (Results 1 – 5 of 5) sorted by relevance
105 TaskType* mutable_task(int i);270 TaskType* Batch<TaskType>::mutable_task(int i) { in mutable_task() function
537 BatchTask& task = *(batch->mutable_task(j)); in SplitOutputTensors()574 batch->mutable_task(i)->status->Update(status); in ProcessFuncBatch()576 batch->mutable_task(i)->context->SetStatus(status); in ProcessFuncBatch()579 batch->mutable_task(i)->done_callback(); in ProcessFuncBatch()701 batch->mutable_task(task_idx)->done_callback(); in ProcessBatch()
64 EXPECT_EQ(task1->size(), batch.mutable_task(1)->size()); in TEST()
143 batch_data.push_back(batch->mutable_task(i)->size()); in TEST()
661 TaskDef* task_def = req.mutable_task(); in AssignTask()