Searched refs:DtypeToPrimitiveType (Results 1 – 4 of 4) sorted by relevance
26 xla::StatusOr<PrimitiveType> DtypeToPrimitiveType(const py::dtype& np_type) { in DtypeToPrimitiveType() function274 auto type_or_status = DtypeToPrimitiveType(array.dtype()); in CastToArray()
450 xla::DtypeToPrimitiveType(aval.attr("dtype"))); in ArgSignatureOfValue()472 TF_ASSIGN_OR_RETURN(dtype, xla::DtypeToPrimitiveType(*to_dtype)); in ArgSignatureOfValue()474 TF_ASSIGN_OR_RETURN(dtype, xla::DtypeToPrimitiveType(raw_dtype)); in ArgSignatureOfValue()485 xla::DtypeToPrimitiveType(numpy_array.dtype())); in ArgSignatureOfValue()519 xla::DtypeToPrimitiveType(h.attr("dtype"))); in ArgSignatureOfValue()
49 StatusOr<PrimitiveType> DtypeToPrimitiveType(const pybind11::dtype& np_type);
191 PrimitiveType type = ValueOrThrow(DtypeToPrimitiveType(dtype)); in BuildXlaCompilerSubmodule()