Home
last modified time | relevance | path

Searched defs:tree_unflatten (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/extension/pytree/
H A Dpybindings.cpp271 static py::object tree_unflatten(py::iterable leaves, py::object o) { in tree_unflatten() function in executorch::extension::pytree::__anon59fa55ac0111::PyTree
276 py::object tree_unflatten(T leaves) const { in tree_unflatten() function in executorch::extension::pytree::__anon59fa55ac0111::PyTree
294 inline py::object tree_unflatten(py::iterable leaves, py::object o) { in tree_unflatten() function
H A D__init__.py32 tree_unflatten as tree_unflatten, unknown
/aosp_15_r20/external/pytorch/torch/utils/
H A D_cxx_pytree.py294 def tree_unflatten(leaves: Iterable[Any], treespec: TreeSpec) -> PyTree: function
H A D_pytree.py878 def tree_unflatten(leaves: Iterable[Any], treespec: TreeSpec) -> PyTree: function