Searched refs:AsyncReduceCUDAWork (Results 1 – 1 of 1) sorted by relevance
1678 class AsyncReduceCUDAWork : public AsyncReduceWork { class1680 AsyncReduceCUDAWork( in AsyncReduceCUDAWork() function in c10d::__anon2d5ad4bf0e11::AsyncReduceCUDAWork1780 work = c10::make_intrusive<AsyncReduceCUDAWork>( in reduce()