Searched refs:DoAllReduce (Results 1 – 1 of 1) sorted by relevance
437 DoAllReduce<xla::S8>(participant); in RunCollectiveOp()441 DoAllReduce<xla::U8>(participant); in RunCollectiveOp()444 DoAllReduce<xla::S32>(participant); in RunCollectiveOp()447 DoAllReduce<xla::U32>(participant); in RunCollectiveOp()450 DoAllReduce<xla::S64>(participant); in RunCollectiveOp()453 DoAllReduce<xla::U64>(participant); in RunCollectiveOp()456 DoAllReduce<xla::F16>(participant); in RunCollectiveOp()459 DoAllReduce<xla::F32>(participant); in RunCollectiveOp()462 DoAllReduce<xla::F64>(participant); in RunCollectiveOp()473 void DoAllReduce(xla::AllReduceParticipantData participant) { in DoAllReduce() function in __anon95669c100311::CpuAllReduceRendezvous