Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Ddispatcher_client.cc47 StatusOr<WorkerHeartbeatResponse> DataServiceDispatcherClient::WorkerHeartbeat( in WorkerHeartbeat()
59 Status DataServiceDispatcherClient::WorkerUpdate( in WorkerUpdate()
77 Status DataServiceDispatcherClient::GetDatasetDef(const std::string& dataset_id, in GetDatasetDef()
92 Status DataServiceDispatcherClient::GetSplit(int64_t iteration_id, in GetSplit()
117 Status DataServiceDispatcherClient::RegisterDataset( in RegisterDataset()
139 Status DataServiceDispatcherClient::GetOrCreateJob( in GetOrCreateJob()
169 Status DataServiceDispatcherClient::GetOrCreateIteration( in GetOrCreateIteration()
188 Status DataServiceDispatcherClient::ReleaseIterationClient( in ReleaseIterationClient()
205 Status DataServiceDispatcherClient::MaybeRemoveTask(int64_t task_id, in MaybeRemoveTask()
224 Status DataServiceDispatcherClient::ClientHeartbeat( in ClientHeartbeat()
[all …]
H A Ddispatcher_client.h41 class DataServiceDispatcherClient : public DataServiceClientBase {
43 DataServiceDispatcherClient(const std::string& address, in DataServiceDispatcherClient() function
H A Dworker_client_test.cc62 dispatcher_client_ = std::make_unique<DataServiceDispatcherClient>( in SetUp()
128 std::unique_ptr<DataServiceDispatcherClient> dispatcher_client_;
H A Ddispatcher_client_test.cc68 dispatcher_client_ = std::make_unique<DataServiceDispatcherClient>( in SetUp()
83 std::unique_ptr<DataServiceDispatcherClient> dispatcher_client_;
H A Dtest_cluster.h148 std::unique_ptr<DataServiceDispatcherClient> dispatcher_client_;
156 dispatcher_client_ = std::make_unique<DataServiceDispatcherClient>( in DatasetClient()
H A Dsplit_provider.cc37 std::make_unique<DataServiceDispatcherClient>(address_, protocol_); in GetNext()
H A Dsplit_provider.h60 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;
H A Dworker_impl.h132 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;
H A Ddata_service_test.cc264 DataServiceDispatcherClient dispatcher(cluster.DispatcherAddress(), "grpc"); in TEST()
H A Dworker_impl.cc167 dispatcher_ = std::make_unique<DataServiceDispatcherClient>( in Start()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/service/
H A Dserver_lib_wrapper.cc110 tensorflow::data::DataServiceDispatcherClient client(address, protocol); in PYBIND11_MODULE()
135 tensorflow::data::DataServiceDispatcherClient client(address, protocol); in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Ddata_service_dataset_op.cc125 DataServiceDispatcherClient client(address, protocol); in GetDataServiceMetadata()
160 DataServiceDispatcherClient client(address, protocol); in GetDataServiceConfig()
442 dispatcher_ = std::make_unique<DataServiceDispatcherClient>( in Initialize()
1242 std::unique_ptr<DataServiceDispatcherClient> dispatcher_;
H A Ddata_service_ops.cc140 DataServiceDispatcherClient client(address, protocol); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/
H A Dsymbols_pybind.txt93 tensorflow::data::DataServiceDispatcherClient::EnsureInitialized
94 tensorflow::data::DataServiceDispatcherClient::GetDataServiceMetadata