Home
last modified time | relevance | path

Searched refs:AsyncSparseAllreduceCUDAWork (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupGloo.cpp1430 class AsyncSparseAllreduceCUDAWork : public AsyncSparseAllreduceWork { class
1432 AsyncSparseAllreduceCUDAWork( in AsyncSparseAllreduceCUDAWork() function in c10d::__anon725d3f3f0911::AsyncSparseAllreduceCUDAWork
1535 work = c10::make_intrusive<AsyncSparseAllreduceCUDAWork>( in allreduce()