Home
last modified time | relevance | path

Searched refs:SubmitParticipant (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc119 StatusOr<std::shared_ptr<BlockingCounter>> SubmitParticipant(
159 StatusOr<std::shared_ptr<BlockingCounter>> Rendezvous::SubmitParticipant( in SubmitParticipant() function in xla::gpu::__anonb6c88d1d0111::Rendezvous
261 rendezvous->SubmitParticipant(participant)); in ExecuteOnStream()
Dnccl_utils.cc342 return NcclCliqueRendezvous::SubmitParticipant( in AcquireNcclClique()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h216 static StatusOr<O> SubmitParticipant( in SubmitParticipant() function
220 TF_ASSIGN_OR_RETURN(auto p, rendezvous->SubmitParticipant(participant)); in SubmitParticipant()
271 SubmitParticipant(const I& participant) { in SubmitParticipant() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc650 TF_CHECK_OK(CpuAllToAllRendezvous::SubmitParticipant( in __xla_cpu_runtime_AllToAll()
700 TF_CHECK_OK(CpuAllReduceRendezvous::SubmitParticipant( in __xla_cpu_runtime_AllReduce()
757 CpuCollectivePermuteRendezvous::SubmitParticipant( in __xla_cpu_runtime_CollectivePermute()