Searched defs:rendezvous_key (Results 1 – 6 of 6) sorted by relevance
125 std::vector<string> rendezvous_key(ctx->num_outputs()); in ComputeAsync() local275 const string& rendezvous_key = Rendezvous::CreateKey( in Compute() local
89 RendezvousKey rendezvous_key(params.run_id, std::move(participants), in ExecuteOnStream() local
218 NcclCliqueRendezvous(const RendezvousKey& rendezvous_key, in NcclCliqueRendezvous()330 const RendezvousKey& rendezvous_key, int local_device_ordinal, in AcquireNcclClique()346 rendezvous_key, [&](const RendezvousKey& rendezvous_key) { in AcquireNcclClique()
632 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllToAll() local670 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllReduce() local742 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_CollectivePermute() local
150 RendezvousKey rendezvous_key; member
344 string rendezvous_key = 2; field