Searched refs:PyNode (Results 1 – 6 of 6) sorted by relevance
27 struct PyNode : public Node { struct28 PyNode(THPObjectPtr obj) : obj(obj.release()) {} in PyNode() function64 ~PyNode() override { in ~PyNode() argument150 std::weak_ptr<torch::autograd::PyNode> cdata;
27 * `PyNode` in `python_function.h`, a subclass of `Node` which forwards30 Outside of `PyNode`, the C++ objects largely avoid referencing Python32 `PyNode`, whose whole point is to let C++ call into Python). And `pyobj`
139 auto PyNode::apply(variable_list&& inputs) -> variable_list { in apply()186 auto PyNode::defer_to_dynamo( in defer_to_dynamo()257 auto PyNode::is_traceable() -> bool { in is_traceable()269 auto PyNode::release_variables() -> void { in release_variables()282 auto PyNode::name() const -> std::string { in name()289 auto PyNode::compiled_autograd_should_lift() const -> bool { in compiled_autograd_should_lift()297 void PyNode::compiled_args(CompiledNodeArgs& args) { in compiled_args()355 variable_list PyNode::apply_with_saved( in apply_with_saved()399 PyObject* PyNode::to_py_args( in to_py_args()433 variable_list PyNode::to_variable_list( in to_variable_list()[all …]
289 if (auto pfw = dynamic_cast<PyNode*>(cdata.get())) { in functionToPyObject()
2258 if (auto py_node_fn = dynamic_cast<PyNode*>(grad_fn.get())) { in THPVariable_subclass_traverse()
610 if (auto pynode = std::dynamic_pointer_cast<PyNode>(call.node)) { in _compiled_autograd_impl()