Searched refs:THCPModule_nccl_all_reduce (Results 1 – 3 of 3) sorted by relevance
10 PyObject* THCPModule_nccl_all_reduce(PyObject* self, PyObject* args);
163 PyObject* THCPModule_nccl_all_reduce(PyObject* self, PyObject* args) { in THCPModule_nccl_all_reduce() function
1902 {"_nccl_all_reduce", THCPModule_nccl_all_reduce, METH_VARARGS, nullptr},