Searched refs:AllLeaves (Results 1 – 2 of 2) sorted by relevance
97 static bool AllLeaves(const pybind11::iterable& x);
184 /*static*/ bool PyTreeDef::AllLeaves(const py::iterable& x) { in AllLeaves() function in xla::PyTreeDef632 pytree.def("all_leaves", &PyTreeDef::AllLeaves); in BuildPytreeSubmodule()