Home
last modified time | relevance | path

Searched defs:torch_C_m (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/csrc/cuda/
DMemPool.cpp12 auto torch_C_m = py::handle(module).cast<py::module>(); in THCPMemPool_init() local
DGraph.cpp24 auto torch_C_m = py::handle(module).cast<py::module>(); in THCPGraph_init() local
/external/pytorch/torch/csrc/distributed/rpc/testing/
Dinit.cpp29 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in faulty_agent_init() local
/external/pytorch/torch/csrc/distributed/autograd/
Dinit.cpp30 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in dist_autograd_init() local
/external/pytorch/torch/csrc/distributed/rpc/
Dinit.cpp43 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in rpc_init() local
/external/pytorch/torch/csrc/distributed/c10d/
Dinit.cpp446 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in c10d_init() local