Searched refs:out_avals (Results 1 – 1 of 1) sorted by relevance
820 std::vector<py::object> out_avals; member1090 cache_entry->out_avals.reserve(avals.size()); in AddCacheEntry()1096 cache_entry->out_avals.push_back(shaped_array); in AddCacheEntry()1183 const std::vector<py::object>& out_avals = cache_entry->out_avals; in Call() local1191 buffer->SetAval(out_avals[i]); in Call()1200 (*device_array)(out_avals[i], sticky_device, out_lazy_exprs[i], in Call()