Searched refs:FlattenUpTo (Results 1 – 2 of 2) sorted by relevance
103 pybind11::list FlattenUpTo(pybind11::handle x) const;
293 py::list PyTreeDef::FlattenUpTo(py::handle xs) const { in FlattenUpTo() function in xla::PyTreeDef636 .def("flatten_up_to", &PyTreeDef::FlattenUpTo) in BuildPytreeSubmodule()