Searched refs:DistributedRuntimeService (Results 1 – 8 of 8) sorted by relevance
40 : public grpc::DistributedRuntimeService::Service {180 class DistributedRuntimeService {182 static xla::StatusOr<std::unique_ptr<DistributedRuntimeService>> Get(188 explicit DistributedRuntimeService(191 ~DistributedRuntimeService();193 DistributedRuntimeService(const DistributedRuntimeService&) = delete;194 DistributedRuntimeService(DistributedRuntimeService&&) = delete;195 DistributedRuntimeService& operator=(const DistributedRuntimeService&) =197 DistributedRuntimeService& operator=(DistributedRuntimeService&&) = delete;
493 xla::StatusOr<std::unique_ptr<DistributedRuntimeService>>494 DistributedRuntimeService::Get( in Get()502 auto service = std::make_unique<DistributedRuntimeService>( in Get()511 DistributedRuntimeService::DistributedRuntimeService( in DistributedRuntimeService() function in xla::DistributedRuntimeService525 DistributedRuntimeService::~DistributedRuntimeService() { Shutdown(); } in ~DistributedRuntimeService()527 void DistributedRuntimeService::Shutdown() { in Shutdown()
26 StatusOr<std::unique_ptr<DistributedRuntimeService>>31 return DistributedRuntimeService::Get(address, credentials, options, in GetDistributedRuntimeService()
36 StatusOr<std::unique_ptr<DistributedRuntimeService>>
68 const std::unique_ptr<grpc::DistributedRuntimeService::Stub> stub_;140 : stub_(grpc::DistributedRuntimeService::NewStub(std::move(channel))), in DistributedRuntimeClientImpl()
122 service DistributedRuntimeService { service
438 py::class_<DistributedRuntimeService, in PYBIND11_MODULE()439 std::unique_ptr<DistributedRuntimeService>> in PYBIND11_MODULE()442 &DistributedRuntimeService::Shutdown, in PYBIND11_MODULE()486 -> StatusOr<std::unique_ptr<DistributedRuntimeService>> { in PYBIND11_MODULE()502 TF_ASSIGN_OR_RETURN(std::unique_ptr<DistributedRuntimeService> service, in PYBIND11_MODULE()
519 class DistributedRuntimeService:534 shutdown_timeout: Optional[int] = ...) -> DistributedRuntimeService: ...