Searched refs:task_def (Results 1 – 5 of 5) sorted by relevance
108 Status DataServiceWorkerImpl::ProcessTaskInternal(const TaskDef& task_def) in ProcessTaskInternal() argument110 std::unique_ptr<Task>& task = tasks_[task_def.task_id()]; in ProcessTaskInternal()113 << task->task_def.task_id(); in ProcessTaskInternal()116 task = absl::make_unique<Task>(task_def); in ProcessTaskInternal()117 VLOG(3) << "Began processing for task " << task_def.task_id() in ProcessTaskInternal()118 << " with processing mode " << task_def.processing_mode(); in ProcessTaskInternal()132 switch (task.task_def.dataset_case()) { in EnsureTaskInitialized()135 params, task.task_def.dataset_def().graph(), &dataset)); in EnsureTaskInitialized()139 Status s = ReadDatasetDef(task.task_def.path(), def); in EnsureTaskInitialized()141 LOG(INFO) << "Failed to read dataset from " << task.task_def.path() in EnsureTaskInitialized()[all …]
61 explicit Task(TaskDef task_def) : task_def(std::move(task_def)) {} in Task()63 TaskDef task_def; member
238 TaskDef* task_def = response->add_new_tasks(); in WorkerHeartbeat() local246 *task_def->mutable_dataset_def() = *dataset_def; in WorkerHeartbeat()250 task_def->set_path(path); in WorkerHeartbeat()252 task_def->set_dataset_id(task->job->dataset_id); in WorkerHeartbeat()253 task_def->set_job_id(task->job->job_id); in WorkerHeartbeat()254 task_def->set_task_id(task->task_id); in WorkerHeartbeat()255 task_def->set_processing_mode( in WorkerHeartbeat()258 task_def->set_num_consumers(task->job->num_consumers.value()); in WorkerHeartbeat()661 TaskDef* task_def = req.mutable_task(); in AssignTask() local662 task_def->set_dataset_id(task->job->dataset_id); in AssignTask()[all …]
85 Status TaskRunner::Create(const TaskDef& task_def, in Create() argument88 if (task_def.optional_num_consumers_case() == TaskDef::kNumConsumers) { in Create()99 task_def.num_consumers()); in Create()
59 static Status Create(const TaskDef& task_def,