Searched refs:NPyBfloat16_Descr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 397 PyArray_Descr NPyBfloat16_Descr = { variable
707 if (PyArray_RegisterCastFunc(&NPyBfloat16_Descr, numpy_type, in RegisterBfloat16Cast()
1349 Py_TYPE(&NPyBfloat16_Descr) = &PyArrayDescr_Type; in Initialize()
1350 npy_bfloat16 = PyArray_RegisterDataType(&NPyBfloat16_Descr); in Initialize()
1358 reinterpret_cast<PyObject*>(&NPyBfloat16_Descr)) < in Initialize()
1420 if (PyArray_RegisterCanCast(&NPyBfloat16_Descr, NPY_FLOAT, NPY_NOSCALAR) < in Initialize()
1424 if (PyArray_RegisterCanCast(&NPyBfloat16_Descr, NPY_DOUBLE, NPY_NOSCALAR) < in Initialize()
1428 if (PyArray_RegisterCanCast(&NPyBfloat16_Descr, NPY_COMPLEX64, NPY_NOSCALAR) < in Initialize()
1432 if (PyArray_RegisterCanCast(&NPyBfloat16_Descr, NPY_COMPLEX128, in Initialize()
|