Searched refs:f_leaf (Results 1 – 2 of 2) sorted by relevance
121 pybind11::handle f_leaf,
438 py::object PyTreeDef::Walk(const py::function& f_node, py::handle f_leaf, in Walk() argument450 agenda.push_back(f_leaf.is_none() ? std::move(leaf) in Walk()451 : f_leaf(std::move(leaf))); in Walk()