Home
last modified time | relevance | path

Searched refs:DistributedRuntimeServiceImpl (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dservice.h32 class DistributedRuntimeServiceImpl final
57 explicit DistributedRuntimeServiceImpl(const Options& options);
58 ~DistributedRuntimeServiceImpl() override;
60 DistributedRuntimeServiceImpl(const DistributedRuntimeServiceImpl&) = delete;
61 DistributedRuntimeServiceImpl(DistributedRuntimeServiceImpl&&) = delete;
62 DistributedRuntimeServiceImpl& operator=(
63 const DistributedRuntimeServiceImpl&) = delete;
64 DistributedRuntimeServiceImpl&& operator=(DistributedRuntimeServiceImpl&&) =
148 const DistributedRuntimeServiceImpl::Options& options);
151 const DistributedRuntimeServiceImpl::Options& options);
[all …]
Dservice.cc28 DistributedRuntimeServiceImpl::DistributedRuntimeServiceImpl( in DistributedRuntimeServiceImpl() function in xla::DistributedRuntimeServiceImpl
35 DistributedRuntimeServiceImpl::~DistributedRuntimeServiceImpl() { in ~DistributedRuntimeServiceImpl()
59 xla::Status DistributedRuntimeServiceImpl::ValidateNodeId(int node_id) { in ValidateNodeId()
72 xla::Status DistributedRuntimeServiceImpl::ValidateSessionId( in ValidateSessionId()
82 ::grpc::Status DistributedRuntimeServiceImpl::Connect( in Connect()
157 ::grpc::Status DistributedRuntimeServiceImpl::Shutdown( in Shutdown()
201 ::grpc::Status DistributedRuntimeServiceImpl::EnumerateDevices( in EnumerateDevices()
256 ::grpc::Status DistributedRuntimeServiceImpl::Heartbeat( in Heartbeat()
281 void DistributedRuntimeServiceImpl::HeartbeatLoop() { in HeartbeatLoop()
310 ::grpc::Status DistributedRuntimeServiceImpl::KeyValueGet( in KeyValueGet()
[all …]
Dclient_server_test.cc35 DistributedRuntimeServiceImpl::Options service_options; in TEST()
37 DistributedRuntimeServiceImpl service(service_options); in TEST()
106 DistributedRuntimeServiceImpl::Options service_options; in TEST()
108 DistributedRuntimeServiceImpl service(service_options); in TEST()
188 DistributedRuntimeServiceImpl::Options service_options; in TEST()
192 DistributedRuntimeServiceImpl service(service_options); in TEST()
237 DistributedRuntimeServiceImpl::Options service_options; in TEST()
241 DistributedRuntimeServiceImpl service(service_options); in TEST()
286 DistributedRuntimeServiceImpl::Options service_options; in TEST()
349 DistributedRuntimeServiceImpl::Options service_options; in TEST()
[all …]
Ddistributed.h38 std::string address, const DistributedRuntimeServiceImpl::Options& options);
Ddistributed.cc27 const DistributedRuntimeServiceImpl::Options& options) { in GetDistributedRuntimeService()