Searched refs:first_participant (Results 1 – 1 of 1) sorted by relevance
488 const xla::AllReduceParticipantData& first_participant = in DoAllReduce() local491 int buffers_per_participant = first_participant.buffers.size(); in DoAllReduce()514 first_participant.buffers[buffer_idx].element_count); in DoAllReduce()520 int element_count = first_participant.buffers[buffer_idx].element_count; in DoAllReduce()