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
135 const RendezvousKey rendezvous_key( in ExecuteOnStream() local
237 NcclCliqueRendezvous(const RendezvousKey& rendezvous_key, in NcclCliqueRendezvous()375 [&](const RendezvousKey& rendezvous_key) { in AcquireNcclClique()
654 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllToAll() local692 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllReduce() local764 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_CollectivePermute() local
231 RendezvousKey rendezvous_key; member
344 string rendezvous_key = 2; field