Searched refs:FlattenInto (Results 1 – 3 of 3) sorted by relevance
92 void FlattenInto(
110 void PyTreeDef::FlattenInto(py::handle handle, std::vector<py::object>& leaves, in FlattenInto() function in xla::PyTreeDef120 FlattenInto(child, leaves, leaf_predicate); in FlattenInto()180 tree->FlattenInto(x, leaves, leaf_predicate); in Flatten()
201 pytree_def.FlattenInto(args[i], arguments.flat_dynamic_args); in ParseArguments()237 arguments.signature.keyword_args[i].value_treedef.FlattenInto( in ParseArguments()