Searched defs:FlattenInto (Results 1 – 1 of 1) sorted by relevance
204 void PyTreeDef::FlattenInto(py::handle handle, in FlattenInto() function in xla::PyTreeDef210 void PyTreeDef::FlattenInto(py::handle handle, std::vector<py::object>& leaves, in FlattenInto() function in xla::PyTreeDef