Searched refs:SafeCastToBfloat16 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 152 bool SafeCastToBfloat16(PyObject* arg, bfloat16* output) { in SafeCastToBfloat16() function
181 if (SafeCastToBfloat16(a, &x) && SafeCastToBfloat16(b, &y)) { in PyBfloat16_Add()
189 if (SafeCastToBfloat16(a, &x) && SafeCastToBfloat16(b, &y)) { in PyBfloat16_Subtract()
197 if (SafeCastToBfloat16(a, &x) && SafeCastToBfloat16(b, &y)) { in PyBfloat16_Multiply()
205 if (SafeCastToBfloat16(a, &x) && SafeCastToBfloat16(b, &y)) { in PyBfloat16_TrueDivide()
288 if (!SafeCastToBfloat16(a, &x) || !SafeCastToBfloat16(b, &y)) { in PyBfloat16_RichCompare()
|