Searched refs:DispatcherState (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/data/service/ |
| D | dispatcher_state_test.cc | 31 using Dataset = DispatcherState::Dataset;
32 using Worker = DispatcherState::Worker;
33 using NamedJobKey = DispatcherState::NamedJobKey;
34 using Job = DispatcherState::Job;
35 using Task = DispatcherState::Task;
39 Status RegisterDataset(int64 id, uint64 fingerprint, DispatcherState& state) { in RegisterDataset()
48 Status RegisterDataset(int64 id, DispatcherState& state) { in RegisterDataset()
52 Status RegisterWorker(std::string worker_address, DispatcherState& state) { in RegisterWorker()
60 DispatcherState& state) { in CreateAnonymousJob()
71 DispatcherState& state) { in CreateNamedJob()
[all …]
|
| D | dispatcher_state.cc | 26 DispatcherState::DispatcherState() {} in DispatcherState() function in tensorflow::data::DispatcherState
28 Status DispatcherState::Apply(const Update& update) { in Apply()
67 void DispatcherState::RegisterDataset( in RegisterDataset()
79 void DispatcherState::RegisterWorker( in RegisterWorker()
89 void DispatcherState::CreateJob(const CreateJobUpdate& create_job) { in CreateJob()
114 void DispatcherState::ProduceSplit(const ProduceSplitUpdate& produce_split) { in ProduceSplit()
127 void DispatcherState::AcquireJobClient( in AcquireJobClient()
139 void DispatcherState::ReleaseJobClient( in ReleaseJobClient()
150 void DispatcherState::CreatePendingTask( in CreatePendingTask()
165 void DispatcherState::ClientHeartbeat( in ClientHeartbeat()
[all …]
|
| D | dispatcher_impl.h | 154 Status RestoreSplitProvider(const DispatcherState::Job& job,
175 absl::optional<DispatcherState::NamedJobKey> named_job_key,
177 std::shared_ptr<const DispatcherState::Job>& job)
184 const std::shared_ptr<const DispatcherState::Job>& job,
190 std::shared_ptr<const DispatcherState::Job> job,
191 std::vector<std::shared_ptr<const DispatcherState::Task>>& tasks)
197 Status CreatePendingTask(std::shared_ptr<const DispatcherState::Job> job,
201 Status CreateTask(std::shared_ptr<const DispatcherState::Job> job,
203 std::shared_ptr<const DispatcherState::Task>& task);
207 std::vector<std::shared_ptr<const DispatcherState::Task>> tasks)
[all …]
|
| D | dispatcher_state.h | 54 class DispatcherState {
56 DispatcherState();
57 DispatcherState(const DispatcherState&) = delete;
58 DispatcherState& operator=(const DispatcherState&) = delete;
|
| D | dispatcher_impl.cc | 69 using Dataset = DispatcherState::Dataset;
70 using Worker = DispatcherState::Worker;
71 using NamedJobKey = DispatcherState::NamedJobKey;
72 using Job = DispatcherState::Job;
73 using Task = DispatcherState::Task;
|