Home
last modified time | relevance | path

Searched refs:DataServiceWorkerClient (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/data/service/
Ddata_service.h144 class DataServiceWorkerClient : public DataServiceClientBase {
146 DataServiceWorkerClient(const std::string& address, in DataServiceWorkerClient() function
179 std::unique_ptr<DataServiceWorkerClient>& out);
Ddata_service.cc335 Status DataServiceWorkerClient::GetElement(const GetElementRequest& req, in GetElement()
341 Status DataServiceWorkerClient::EnsureInitialized() { in EnsureInitialized()
351 void DataServiceWorkerClient::TryCancel() { client_->TryCancel(); } in TryCancel()
366 std::unique_ptr<DataServiceWorkerClient>& out) { in CreateDataServiceWorkerClient()
367 auto client = absl::make_unique<DataServiceWorkerClient>(address, protocol, in CreateDataServiceWorkerClient()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddata_service_dataset_op.cc382 std::unique_ptr<DataServiceWorkerClient> worker) in Task()
387 const std::unique_ptr<DataServiceWorkerClient> worker;
467 std::unique_ptr<DataServiceWorkerClient> worker; in AddTask()