Home
last modified time | relevance | path

Searched refs:functionToPyObject (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/
H A Dinit.cpp54 torch::autograd::functionToPyObject( in dist_autograd_init()
72 torch::autograd::functionToPyObject( in dist_autograd_init()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_cpp_function.cpp144 PyObject* py_fn = functionToPyObject(c_tuple.function); in THPCppFunction_next_functions()
284 PyObject* functionToPyObject(const std::shared_ptr<Node>& cdata) { in functionToPyObject() function
H A Dpython_anomaly_mode.cpp74 THPObjectPtr parent_node_(functionToPyObject(parent_node)); in assign_parent()
H A Dpython_cpp_function.h104 PyObject* functionToPyObject(const std::shared_ptr<Node>& cdata);
H A Dpython_function.cpp1615 PyObject* fn = functionToPyObject(edge.function); in THPFunction_next_functions()
H A Dpython_variable.cpp1004 return functionToPyObject(var.grad_fn()); in THPVariable_get_grad_fn()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1235 torch::autograd::functionToPyObject(nodes[i]->getptr()); in THPModule_getCurrentGraphTaskExecutionOrder()
1250 return torch::autograd::functionToPyObject( in THPModule_getCurrentNode()