Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc292 const py::dtype* DtypeTo32BitDtype(const py::dtype& dtype) { in DtypeTo32BitDtype() function
468 const py::dtype* to_dtype = DtypeTo32BitDtype(raw_dtype); in ArgSignatureOfValue()
723 const py::dtype* to_dtype = DtypeTo32BitDtype(numpy_array.dtype()); in HandleNdarray()
1278 const py::dtype* res = DtypeTo32BitDtype(dtype); in BuildJaxjitSubmodule()