Searched refs:f_node (Results 1 – 3 of 3) sorted by relevance
120 pybind11::object Walk(const pybind11::function& f_node,
438 py::object PyTreeDef::Walk(const py::function& f_node, py::handle f_leaf, in Walk() argument470 agenda.push_back(f_node(tuple)); in Walk()
2329 auto* f_node = graph_view.GetNode("f"); in TEST_F() local2330 ASSERT_NE(f_node, nullptr); in TEST_F()2333 {{e_node, f_node}})); in TEST_F()