Searched refs:torch_C_module (Results 1 – 5 of 5) sorted by relevance
24 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in faulty_agent_init() local25 if (!torch_C_module) { in faulty_agent_init()29 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in faulty_agent_init()
25 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in dist_autograd_init() local26 if (!torch_C_module) { in dist_autograd_init()30 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in dist_autograd_init()
118 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in THPAutograd_initExtension() local119 if (!torch_C_module) in THPAutograd_initExtension()121 auto _C_m = py::handle(torch_C_module).cast<py::module>(); in THPAutograd_initExtension()876 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in set_grad_enabled() local878 r, args, kwargs, torch_C_module, "torch._C", "_set_grad_enabled"); in set_grad_enabled()930 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in set_multithreading_enabled() local935 torch_C_module, in set_multithreading_enabled()968 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in set_view_replay_enabled() local973 torch_C_module, in set_view_replay_enabled()
38 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in rpc_init() local39 if (!torch_C_module) { in rpc_init()43 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in rpc_init()
441 auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C")); in c10d_init() local442 if (!torch_C_module) { in c10d_init()446 auto torch_C_m = py::handle(torch_C_module).cast<py::module>(); in c10d_init()