Searched defs:recvbuff (Results 1 – 2 of 2) sorted by relevance
738 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches() local759 void* recvbuff = nullptr; in LoopKernelLaunches() local796 void* recvbuff = in LoopKernelLaunches() local812 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches() local
826 auto* recvbuff = reinterpret_cast<char*>(output.data_ptr()); in all2all_single_equal_split() local859 void* recvbuff, in all2all_single_unequal_split()1054 auto* recvbuff = reinterpret_cast<char*>(outputs[r].data_ptr()); in gather() local1115 auto* recvbuff = reinterpret_cast<char*>(outputs.data_ptr()); in scatter() local