Home
last modified time | relevance | path

Searched defs:out_tree (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc182 auto out_tree = py::cast<xla::PyTreeDef>(pmap_data.attr("out_pytree_def")); in AddCacheEntry() local
Djax_jit.cc711 auto out_tree = py::cast<xla::PyTreeDef>(executable_handlers_out_tree[1]); in PopulateCacheEntry() local