Searched refs:FloatToBFloat16 (Results 1 – 5 of 5) sorted by relevance
49 static inline tensorflow::bfloat16 FloatToBFloat16(float float_val) { in FloatToBFloat16() function71 return FloatToBFloat16(NumTraits<float>::highest());75 return FloatToBFloat16(NumTraits<float>::lowest());79 return FloatToBFloat16(NumTraits<float>::infinity());83 return FloatToBFloat16(NumTraits<float>::quiet_NaN());
133 FloatToBFloat16(a, b, 100); in TEST()165 FloatToBFloat16(inp, out, N); in BM_FloatToBFloat16()
58 void FloatToBFloat16(const float* src, bfloat16* dst, int64 size);
20 void FloatToBFloat16(const float* src, bfloat16* dst, int64 size) { in FloatToBFloat16() function
514 FloatToBFloat16(out_float.flat<float>().data(), in Compute()