Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/data/service/
Ddata_service.cc60 Status DataServiceDispatcherClient::WorkerHeartbeat( in WorkerHeartbeat()
86 Status DataServiceDispatcherClient::WorkerUpdate( in WorkerUpdate()
104 Status DataServiceDispatcherClient::GetDatasetDef(int64 dataset_id, in GetDatasetDef()
119 Status DataServiceDispatcherClient::GetSplit(int64 job_id, int64 repetition, in GetSplit()
141 Status DataServiceDispatcherClient::RegisterDataset(GraphDef dataset, in RegisterDataset()
156 Status DataServiceDispatcherClient::GetOrCreateJob( in GetOrCreateJob()
183 Status DataServiceDispatcherClient::ReleaseJobClient(int64 job_client_id) { in ReleaseJobClient()
198 Status DataServiceDispatcherClient::ClientHeartbeat( in ClientHeartbeat()
209 Status DataServiceDispatcherClient::GetWorkers( in GetWorkers()
226 Status DataServiceDispatcherClient::EnsureInitialized() { in EnsureInitialized()
[all …]
Ddata_service.h79 class DataServiceDispatcherClient : public DataServiceClientBase {
81 DataServiceDispatcherClient(const std::string& address, in DataServiceDispatcherClient() function
173 std::unique_ptr<DataServiceDispatcherClient>& out);
Dsplit_provider.cc29 absl::make_unique<DataServiceDispatcherClient>(address_, protocol_); in GetNext()
Ddata_service_test.cc70 DataServiceDispatcherClient dispatcher(cluster.DispatcherAddress(), in TEST()
Dsplit_provider.h53 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;
Dworker_impl.h86 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;
Dworker_impl.cc73 dispatcher_ = absl::make_unique<DataServiceDispatcherClient>( in Start()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddata_service_ops.cc61 DataServiceDispatcherClient client(address, protocol); in Compute()
Ddata_service_dataset_op.cc263 dispatcher_ = absl::make_unique<DataServiceDispatcherClient>( in Initialize()
830 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;