Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dpython_nccl.h9 PyObject* THCPModule_nccl_reduce(PyObject* self, PyObject* args);
H A Dpython_nccl.cpp130 PyObject* THCPModule_nccl_reduce(PyObject* self, PyObject* args) { in THCPModule_nccl_reduce() function
H A DModule.cpp1901 {"_nccl_reduce", THCPModule_nccl_reduce, METH_VARARGS, nullptr},