Searched defs:out_tree (Results 1 – 2 of 2) sorted by relevance
182 auto out_tree = py::cast<xla::PyTreeDef>(pmap_data.attr("out_pytree_def")); in AddCacheEntry() local
711 auto out_tree = py::cast<xla::PyTreeDef>(executable_handlers_out_tree[1]); in PopulateCacheEntry() local