Home
last modified time | relevance | path

Searched refs:out_pytree_def (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc114 xla::PyTreeDef out_pytree_def; member
249 cache_entry->out_pytree_def = std::move(out_tree); in AddCacheEntry()
336 return cache_entry->out_pytree_def.Unflatten(flat_sharded_device_arrays); in Call()
Djax_jit.cc816 xla::PyTreeDef out_pytree_def; member
1082 cache_entry->out_pytree_def = std::move(out_tree); in AddCacheEntry()
1204 return cache_entry->out_pytree_def.Unflatten(flat_device_arrays); in Call()