Home
last modified time | relevance | path

Searched defs:DistributedRuntimeService (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dprotocol.proto122 service DistributedRuntimeService { service
126 rpc Connect(ConnectRequest) returns (ConnectResponse) {}
131 rpc EnumerateDevices(EnumerateDevicesRequest)
137 rpc Heartbeat(HeartbeatRequest) returns (HeartbeatResponse) {}
142 rpc Shutdown(ShutdownRequest) returns (ShutdownResponse) {}
150 rpc KeyValueGet(KeyValueGetRequest) returns (KeyValueGetResponse) {}
153 rpc KeyValueSet(KeyValueSetRequest) returns (KeyValueSetResponse) {}
156 rpc WaitAtBarrier(WaitAtBarrierRequest) returns (WaitAtBarrierResponse) {}
H A Dservice.cc511 DistributedRuntimeService::DistributedRuntimeService( in DistributedRuntimeService() function in xla::DistributedRuntimeService