Searched defs:output_buff (Results 1 – 3 of 3) sorted by relevance
64 T *output_buff = reinterpret_cast<T *>(recvbuff); in RingAllGather() local77 …ollectiveOpsImpl::RingAllGatherImpl(uint32_t send_to_rank, uint32_t recv_from_rank, T *output_buff, in RingAllGatherImpl()
101 auto *output_buff = reinterpret_cast<float *>(output_data); in RingAllReduce() local190 float *output_buff = reinterpret_cast<float *>(output_data); in ReduceBroadcastAllReduce() local
70 T *output_buff = reinterpret_cast<T *>(recvbuff); in RingAllReduce() local85 const std::vector<size_t> &chunk_offset, T *output_buff) { in RunRingAllReduce()206 T *output_buff = reinterpret_cast<T *>(recvbuff); in ReduceBroadcastAllReduce() local309 T *output_buff = reinterpret_cast<T *>(recvbuff); in RingAllGather() local