Home
last modified time | relevance | path

Searched defs:WaitAtBarrier (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dclient.cc350 xla::Status DistributedRuntimeClientImpl::WaitAtBarrier( in WaitAtBarrier() function in xla::DistributedRuntimeClientImpl
544 xla::Status DistributedRuntimeCoordinationServiceClient::WaitAtBarrier( in WaitAtBarrier() function in xla::DistributedRuntimeCoordinationServiceClient
H A Dprotocol.proto156 rpc WaitAtBarrier(WaitAtBarrierRequest) returns (WaitAtBarrierResponse) {} rpc
H A Dservice.cc406 ::grpc::Status DistributedRuntimeServiceImpl::WaitAtBarrier( in WaitAtBarrier() function in xla::DistributedRuntimeServiceImpl
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/coordination/
H A Dcoordination_service_agent.cc672 Status CoordinationServiceAgentImpl::WaitAtBarrier( in WaitAtBarrier() function in tensorflow::__anonca10de4b0111::CoordinationServiceAgentImpl