Searched refs:AsyncSparseAllreduceWork (Results 1 – 1 of 1) sorted by relevance
1102 class AsyncSparseAllreduceWork : public ProcessGroupGloo::AsyncWork { class1104 AsyncSparseAllreduceWork( in AsyncSparseAllreduceWork() function in c10d::__anon725d3f3f0911::AsyncSparseAllreduceWork1430 class AsyncSparseAllreduceCUDAWork : public AsyncSparseAllreduceWork {1437 : AsyncSparseAllreduceWork(context, inputs, tag, seq) { in AsyncSparseAllreduceCUDAWork()1525 work = c10::make_intrusive<AsyncSparseAllreduceWork>( in allreduce()