Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/lib/core/
Dbfloat16.cc389 int npy_bfloat16_ = -1; variable
477 if (PyArray_RegisterCastFunc(PyArray_DescrFromType(numpy_type), npy_bfloat16_, in RegisterBfloat16Cast()
564 npy_bfloat16_ = PyArray_RegisterDataType(&NPyBfloat16_Descr); in Initialize()
565 if (npy_bfloat16_ < 0) return false; in Initialize()
621 if (PyUFunc_RegisterLoopForType(ufunc, npy_bfloat16_, fn, in Initialize()
631 {npy_bfloat16_, npy_bfloat16_, NPY_BOOL}}; in Initialize()
662 if (npy_bfloat16_ >= 0) { in RegisterNumpyBfloat16()
681 CHECK_GE(npy_bfloat16_, 0); in Bfloat16NumpyType()
682 return npy_bfloat16_; in Bfloat16NumpyType()