Home
last modified time | relevance | path

Searched defs:task_def (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dauto_shard_rewriter_test.cc74 TaskDef task_def; in GetTaskDef() local
82 TaskDef task_def = GetTaskDef(ProcessingModeDef::FILE_OR_DATA, in TEST() local
98 TaskDef task_def = GetTaskDef(ProcessingModeDef::DATA, /*num_workers=*/3, in TEST() local
114 TaskDef task_def = GetTaskDef(ProcessingModeDef::FILE, /*num_workers=*/3, in TEST() local
126 TaskDef task_def = GetTaskDef(ProcessingModeDef::HINT, /*num_workers=*/3, in TEST() local
142 TaskDef task_def = in TEST() local
153 TaskDef task_def = in TEST() local
170 TaskDef task_def = in TEST() local
183 TaskDef task_def = in TEST() local
194 TaskDef task_def = in TEST() local
H A Dworker_impl.h92 TaskDef task_def; member
H A Dauto_shard_rewriter.cc87 StatusOr<AutoShardRewriter> AutoShardRewriter::Create(const TaskDef& task_def) { in Create()
H A Ddispatcher_impl.cc326 TaskDef* task_def = response->add_new_tasks(); in FindNewTasks() local
908 TaskDef* task_def = req.mutable_task(); in AssignTask() local
H A Dtask_runner.cc66 const TaskDef& task_def, in Create()
H A Dworker_impl.cc292 Status DataServiceWorkerImpl::ProcessTaskInternal(const TaskDef& task_def) in ProcessTaskInternal()