Home
last modified time | relevance | path

Searched refs:job_client_id (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_state.cc129 int64 job_client_id = acquire_job_client.job_client_id(); in AcquireJobClient() local
130 std::shared_ptr<Job>& job = jobs_for_client_ids_[job_client_id]; in AcquireJobClient()
136 std::max(next_available_job_client_id_, job_client_id + 1); in AcquireJobClient()
141 int64 job_client_id = release_job_client.job_client_id(); in ReleaseJobClient() local
142 std::shared_ptr<Job>& job = jobs_for_client_ids_[job_client_id]; in ReleaseJobClient()
147 jobs_for_client_ids_.erase(job_client_id); in ReleaseJobClient()
167 int64 job_client_id = client_heartbeat.job_client_id(); in ClientHeartbeat() local
168 auto& job = jobs_for_client_ids_[job_client_id]; in ClientHeartbeat()
177 task.ready_consumers.insert(job_client_id); in ClientHeartbeat()
295 Status DispatcherState::JobForJobClientId(int64 job_client_id, in JobForJobClientId() argument
[all …]
Ddispatcher_impl.cc440 int64 job_client_id; in GetOrCreateJob() local
441 TF_RETURN_IF_ERROR(AcquireJobClientId(job, job_client_id)); in GetOrCreateJob()
442 response->set_job_client_id(job_client_id); in GetOrCreateJob()
459 int64 job_client_id; in GetOrCreateJob() local
460 TF_RETURN_IF_ERROR(AcquireJobClientId(job, job_client_id)); in GetOrCreateJob()
461 response->set_job_client_id(job_client_id); in GetOrCreateJob()
475 int64 job_client_id = request->job_client_id(); in ReleaseJobClient() local
477 TF_RETURN_IF_ERROR(state_.JobForJobClientId(job_client_id, job)); in ReleaseJobClient()
481 release_job_client->set_job_client_id(job_client_id); in ReleaseJobClient()
559 const std::shared_ptr<const Job>& job, int64& job_client_id) in AcquireJobClientId() argument
[all …]
Djournal.proto62 int64 job_client_id = 2; field
66 int64 job_client_id = 1; field
80 int64 job_client_id = 1; field
Ddata_service.cc159 int64& job_client_id) { in GetOrCreateJob() argument
179 job_client_id = resp.job_client_id(); in GetOrCreateJob()
183 Status DataServiceDispatcherClient::ReleaseJobClient(int64 job_client_id) { in ReleaseJobClient() argument
186 req.set_job_client_id(job_client_id); in ReleaseJobClient()
192 absl::StrCat("Failed to release job client with id ", job_client_id), in ReleaseJobClient()
Ddispatcher_state_test.cc84 Status AcquireJobClientId(int64 job_id, int64 job_client_id, in AcquireJobClientId() argument
90 acquire_job_client->set_job_client_id(job_client_id); in AcquireJobClientId()
95 Status ReleaseJobClientId(int64 job_client_id, int64 release_time, in ReleaseJobClientId() argument
100 release_job_client->set_job_client_id(job_client_id); in ReleaseJobClientId()
462 int64 job_client_id = 6; in TEST() local
467 TF_EXPECT_OK(AcquireJobClientId(job_id, job_client_id, state)); in TEST()
468 TF_EXPECT_OK(ReleaseJobClientId(job_client_id, release_time, state)); in TEST()
472 Status s = state.JobForJobClientId(job_client_id, job); in TEST()
Ddata_service.h118 int64& job_client_id);
122 Status ReleaseJobClient(int64 job_client_id);
Ddispatcher.proto94 int64 job_client_id = 1; field
98 int64 job_client_id = 1; field
105 int64 job_client_id = 1; field
Ddispatcher_impl.h185 int64& job_client_id) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);
Ddispatcher_state.h194 Status JobForJobClientId(int64 job_client_id,