Searched refs:npy_bfloat16 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 55 int npy_bfloat16 = NPY_NOTYPE; variable
138 if (PyArray_TYPE(arr) != npy_bfloat16) { in CastToBfloat16()
139 ref = make_safe(PyArray_Cast(arr, npy_bfloat16)); in CastToBfloat16()
273 if (PyArray_TYPE(arr) != npy_bfloat16) { in PyBfloat16_New()
274 return PyArray_Cast(arr, npy_bfloat16); in PyBfloat16_New()
584 static int Dtype() { return npy_bfloat16; } in Dtype()
704 if (PyArray_RegisterCastFunc(descr, npy_bfloat16, NPyCast<T, bfloat16>) < 0) { in RegisterBfloat16Cast()
819 if (PyUFunc_RegisterLoopForType(ufunc, npy_bfloat16, fn, in RegisterUFunc()
883 return {npy_bfloat16, npy_bfloat16, npy_bfloat16, npy_bfloat16}; in Types()
1323 npy_bfloat16 = typenum; in Initialize()
[all …]
|