Searched defs:leaf_predicate (Results 1 – 1 of 1) sorted by relevance
128 const absl::optional<py::function>& leaf_predicate) { in FlattenIntoImpl()206 absl::optional<py::function> leaf_predicate) { in FlattenInto()211 absl::optional<py::function> leaf_predicate) { in FlattenInto()216 PyTreeDef::Flatten(py::handle x, absl::optional<py::function> leaf_predicate) { in Flatten()