Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc823 std::vector<py::object> out_lazy_exprs; member
1091 cache_entry->out_lazy_exprs.reserve(avals.size()); in AddCacheEntry()
1098 cache_entry->out_lazy_exprs.push_back(lazy_expr); in AddCacheEntry()
1184 const std::vector<py::object>& out_lazy_exprs = cache_entry->out_lazy_exprs; in Call() local
1190 if (out_lazy_exprs[i].is_none()) { // No LazyExpr. in Call()
1200 (*device_array)(out_avals[i], sticky_device, out_lazy_exprs[i], in Call()