Home
last modified time | relevance | path

Searched defs:shaped_array (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc1093 py::object shaped_array = py::reinterpret_borrow<py::object>(avals[i]); in AddCacheEntry() local
1249 static const auto* shaped_array = in BuildJaxjitSubmodule() local