Searched refs:DistributedRuntimeService (Results 1 – 9 of 9) sorted by relevance
33 : public grpc::DistributedRuntimeService::Service {143 class DistributedRuntimeService {145 static xla::StatusOr<std::unique_ptr<DistributedRuntimeService>> Get(150 explicit DistributedRuntimeService(152 ~DistributedRuntimeService();154 DistributedRuntimeService(const DistributedRuntimeService&) = delete;155 DistributedRuntimeService(DistributedRuntimeService&&) = delete;156 DistributedRuntimeService& operator=(const DistributedRuntimeService&) =158 DistributedRuntimeService& operator=(DistributedRuntimeService&&) = delete;
355 xla::StatusOr<std::unique_ptr<DistributedRuntimeService>>356 DistributedRuntimeService::Get( in Get()360 auto service = absl::make_unique<DistributedRuntimeService>(options); in Get()373 DistributedRuntimeService::DistributedRuntimeService( in DistributedRuntimeService() function in xla::DistributedRuntimeService377 DistributedRuntimeService::~DistributedRuntimeService() { in ~DistributedRuntimeService()
24 StatusOr<std::unique_ptr<DistributedRuntimeService>>29 return DistributedRuntimeService::Get(address, credentials, options); in GetDistributedRuntimeService()
36 StatusOr<std::unique_ptr<DistributedRuntimeService>>
122 const std::unique_ptr<grpc::DistributedRuntimeService::Stub> stub_;
110 service DistributedRuntimeService { service
32 : stub_(grpc::DistributedRuntimeService::NewStub(std::move(channel))), in DistributedRuntimeClient()
297 DistributedRuntimeService::Get(absl::StrCat("[::]:", port), in TEST()
376 py::class_<DistributedRuntimeService, in PYBIND11_MODULE()377 std::unique_ptr<DistributedRuntimeService>> in PYBIND11_MODULE()