Home
last modified time | relevance | path

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

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