Searched refs:DistributedRuntimeServiceImpl (Results 1 – 6 of 6) sorted by relevance
39 class DistributedRuntimeServiceImpl final64 explicit DistributedRuntimeServiceImpl(const Options& options);65 ~DistributedRuntimeServiceImpl() override;67 DistributedRuntimeServiceImpl(const DistributedRuntimeServiceImpl&) = delete;68 DistributedRuntimeServiceImpl(DistributedRuntimeServiceImpl&&) = delete;69 DistributedRuntimeServiceImpl& operator=(70 const DistributedRuntimeServiceImpl&) = delete;71 DistributedRuntimeServiceImpl&& operator=(DistributedRuntimeServiceImpl&&) =160 CoordinationServiceImpl(const DistributedRuntimeServiceImpl::Options& options,185 const DistributedRuntimeServiceImpl::Options& options,[all …]
45 const xla::DistributedRuntimeServiceImpl::Options& options) { in EnableCoordinationService()79 DistributedRuntimeServiceImpl::DistributedRuntimeServiceImpl( in DistributedRuntimeServiceImpl() function in xla::DistributedRuntimeServiceImpl86 DistributedRuntimeServiceImpl::~DistributedRuntimeServiceImpl() { in ~DistributedRuntimeServiceImpl()110 xla::Status DistributedRuntimeServiceImpl::ValidateNodeId(int node_id) { in ValidateNodeId()123 xla::Status DistributedRuntimeServiceImpl::ValidateSessionId( in ValidateSessionId()133 ::grpc::Status DistributedRuntimeServiceImpl::Connect( in Connect()208 ::grpc::Status DistributedRuntimeServiceImpl::Shutdown( in Shutdown()252 ::grpc::Status DistributedRuntimeServiceImpl::EnumerateDevices( in EnumerateDevices()307 ::grpc::Status DistributedRuntimeServiceImpl::Heartbeat( in Heartbeat()332 void DistributedRuntimeServiceImpl::HeartbeatLoop() { in HeartbeatLoop()[all …]
66 DistributedRuntimeServiceImpl::Options service_options = {}, in StartService()89 std::make_unique<DistributedRuntimeServiceImpl>(service_options);112 std::unique_ptr<DistributedRuntimeServiceImpl> distributed_runtime_service_;479 DistributedRuntimeServiceImpl::Options service_options; in TEST_P()
38 std::string address, const DistributedRuntimeServiceImpl::Options& options,
28 std::string address, const DistributedRuntimeServiceImpl::Options& options, in GetDistributedRuntimeService()
487 DistributedRuntimeServiceImpl::Options options; in PYBIND11_MODULE()