Home
last modified time | relevance | path

Searched refs:DistributedRuntimeClient (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dclient_server_test.cc49 DistributedRuntimeClient::Options client_options; in TEST()
51 DistributedRuntimeClient client( in TEST()
136 DistributedRuntimeClient::Options client_options; in TEST()
138 DistributedRuntimeClient client( in TEST()
154 DistributedRuntimeClient::Options client_options; in TEST()
156 DistributedRuntimeClient client( in TEST()
198 DistributedRuntimeClient::Options client_options; in TEST()
205 DistributedRuntimeClient client( in TEST()
247 DistributedRuntimeClient::Options client_options; in TEST()
257 DistributedRuntimeClient client( in TEST()
[all …]
Dclient.cc30 DistributedRuntimeClient::DistributedRuntimeClient( in DistributedRuntimeClient() function in xla::DistributedRuntimeClient
35 DistributedRuntimeClient::~DistributedRuntimeClient() { in ~DistributedRuntimeClient()
55 /*static*/ absl::string_view DistributedRuntimeClient::StateToString( in StateToString()
69 xla::Status DistributedRuntimeClient::Connect() { in Connect()
133 xla::Status DistributedRuntimeClient::EnumerateDevices( in EnumerateDevices()
162 xla::Status DistributedRuntimeClient::Shutdown() { in Shutdown()
193 xla::StatusOr<std::string> DistributedRuntimeClient::BlockingKeyValueGet( in BlockingKeyValueGet()
219 xla::Status DistributedRuntimeClient::KeyValueSet(std::string key, in KeyValueSet()
241 void DistributedRuntimeClient::HeartbeatLoop() { in HeartbeatLoop()
Dclient.h32 class DistributedRuntimeClient {
90 DistributedRuntimeClient(std::shared_ptr<::grpc::Channel> channel,
92 explicit DistributedRuntimeClient(std::shared_ptr<::grpc::Channel> channel) in DistributedRuntimeClient() function
93 : DistributedRuntimeClient(channel, Options()) {} in DistributedRuntimeClient()
94 ~DistributedRuntimeClient();
Ddistributed.cc32 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient( in GetDistributedRuntimeClient()
33 std::string address, const DistributedRuntimeClient::Options& options) { in GetDistributedRuntimeClient()
38 return absl::make_unique<DistributedRuntimeClient>(channel, options); in GetDistributedRuntimeClient()
Ddistributed.h42 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient(
43 std::string address, const DistributedRuntimeClient::Options& options);
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc246 std::shared_ptr<DistributedRuntimeClient> distributed_client, in PYBIND11_MODULE()
379 py::class_<DistributedRuntimeClient, in PYBIND11_MODULE()
380 std::shared_ptr<DistributedRuntimeClient>> in PYBIND11_MODULE()
382 distributed_runtime_client.def("connect", &DistributedRuntimeClient::Connect) in PYBIND11_MODULE()
383 .def("shutdown", &DistributedRuntimeClient::Shutdown); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc178 NcclIdStore(int node_id, std::shared_ptr<DistributedRuntimeClient> client, in NcclIdStore()
188 const std::shared_ptr<DistributedRuntimeClient> client_;
246 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in BuildDistributedDevices()
315 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id) { in GetGpuClient()
Dgpu_device.h59 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id);