Searched refs:shaped_array (Results 1 – 1 of 1) sorted by relevance
1093 py::object shaped_array = py::reinterpret_borrow<py::object>(avals[i]); in AddCacheEntry() local1096 cache_entry->out_avals.push_back(shaped_array); in AddCacheEntry()1249 static const auto* shaped_array = in BuildJaxjitSubmodule() local1251 buffer->SetAval((*shaped_array)( in BuildJaxjitSubmodule()