Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dtypes.cc53 xla::StatusOr<py::dtype> PrimitiveTypeToDtype(PrimitiveType type) { in PrimitiveTypeToDtype() function
208 PrimitiveTypeToDtype(m.shape().element_type())); in LiteralToPython()
Dpy_buffer.cc64 return PrimitiveTypeToDtype(primitive).ValueOrDie(); in python_dtype()
117 PrimitiveTypeToDtype(buffer_->on_device_shape().element_type())); in AsNumPyArray()
Dxla_compiler.cc210 return ValueOrThrow(PrimitiveTypeToDtype(shape.element_type())); in BuildXlaCompilerSubmodule()
217 return ValueOrThrow(PrimitiveTypeToDtype(shape.element_type())); in BuildXlaCompilerSubmodule()
Dxla.cc295 return PrimitiveTypeToDtype(primitive).ValueOrDie(); in PYBIND11_MODULE()
Dtypes.h52 StatusOr<pybind11::dtype> PrimitiveTypeToDtype(PrimitiveType type);
Djax_jit.cc1265 return PrimitiveTypeToDtype(sig.dtype); in BuildJaxjitSubmodule()