Searched refs:out_tree (Results 1 – 2 of 2) sorted by relevance
248 auto out_tree = py::cast<xla::PyTreeDef>(pmap_data["out_pytree_def"]); in AddCacheEntry() local249 cache_entry->out_pytree_def = std::move(out_tree); in AddCacheEntry()
1081 auto out_tree = py::cast<xla::PyTreeDef>(executable_handlers_out_tree[1]); in AddCacheEntry() local1082 cache_entry->out_pytree_def = std::move(out_tree); in AddCacheEntry()