Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc1062 py::tuple executable_handlers_out_tree = py::cast<py::tuple>(tuple[1]); in AddCacheEntry() local
1063 if (executable_handlers_out_tree.size() != 5) { in AddCacheEntry()
1068 executable_handlers_out_tree.size(), " where provided: ", in AddCacheEntry()
1070 py::str(py::repr(executable_handlers_out_tree))))); in AddCacheEntry()
1074 executable_handlers_out_tree[0]); in AddCacheEntry()
1081 auto out_tree = py::cast<xla::PyTreeDef>(executable_handlers_out_tree[1]); in AddCacheEntry()
1085 py::cast<py::object>(executable_handlers_out_tree[2]); in AddCacheEntry()
1086 auto avals = py::cast<py::list>(executable_handlers_out_tree[3]); in AddCacheEntry()
1087 auto lazy_exprs = py::cast<py::list>(executable_handlers_out_tree[4]); in AddCacheEntry()