Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dpython_nccl.h13 PyObject* THCPModule_nccl_reduce_scatter(PyObject* self, PyObject* args);
H A Dpython_nccl.cpp259 PyObject* THCPModule_nccl_reduce_scatter(PyObject* self, PyObject* args) { in THCPModule_nccl_reduce_scatter() function
H A DModule.cpp1906 THCPModule_nccl_reduce_scatter,