Searched defs:segment_recv (Results 1 – 2 of 2) sorted by relevance
186 T* segment_recv = (T*)temp->tensor_data().data(); in RingAllreduce() local237 T* segment_recv = &(buffer[segment_starts[recv_seg_id]]); in RingAllreduce() local309 T* segment_recv = &(buffer[offset_recv]); in RingAllgather() local