Home
last modified time | relevance | path

Searched refs:named_job_key (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_state.cc91 absl::optional<NamedJobKey> named_job_key; in CreateJob() local
93 named_job_key.emplace(create_job.named_job_key().name(), in CreateJob()
94 create_job.named_job_key().index()); in CreateJob()
103 named_job_key, num_consumers); in CreateJob()
107 if (named_job_key.has_value()) { in CreateJob()
108 DCHECK(!named_jobs_.contains(named_job_key.value())); in CreateJob()
109 named_jobs_[named_job_key.value()] = job; in CreateJob()
280 Status DispatcherState::NamedJobByKey(NamedJobKey named_job_key, in NamedJobByKey() argument
282 auto it = named_jobs_.find(named_job_key); in NamedJobByKey()
284 return errors::NotFound("Named job key (", named_job_key.name, ", ", in NamedJobByKey()
[all …]
Ddispatcher_state.h127 absl::optional<NamedJobKey> named_job_key, in Job()
132 named_job_key(named_job_key), in Job()
142 const absl::optional<NamedJobKey> named_job_key; member
Ddispatcher_state_test.cc70 Status CreateNamedJob(int64 job_id, int64 dataset_id, NamedJobKey named_job_key, in CreateNamedJob() argument
78 key->set_name(named_job_key.name); in CreateNamedJob()
79 key->set_index(named_job_key.index); in CreateNamedJob()
233 NamedJobKey named_job_key("test", 1); in TEST() local
234 TF_EXPECT_OK(CreateNamedJob(job_id, dataset_id, named_job_key, state)); in TEST()
236 TF_EXPECT_OK(state.NamedJobByKey(named_job_key, job)); in TEST()
Ddispatcher_impl.cc491 DCHECK(job->named_job_key.has_value()); in ValidateMatchingJob()
492 std::string job_name = job->named_job_key->name; in ValidateMatchingJob()
508 absl::optional<NamedJobKey> named_job_key, in CreateJob() argument
528 if (named_job_key.has_value()) { in CreateJob()
530 key->set_name(named_job_key->name); in CreateJob()
531 key->set_index(named_job_key->index); in CreateJob()
Djournal.proto45 NamedJobKeyDef named_job_key = 4; field
Ddispatcher_impl.h175 absl::optional<DispatcherState::NamedJobKey> named_job_key,