Searched refs:FloatToBFloat16 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | bfloat16_test.cc | 33 FloatToBFloat16(a, b, 100); in TEST()
49 FloatToBFloat16(inp, out, N); in BM_FloatToBFloat16()
|
| D | bfloat16.h | 55 void FloatToBFloat16(const float* src, bfloat16* dst, int64 size);
|
| D | bfloat16.cc | 28 void FloatToBFloat16(const float* src, bfloat16* dst, int64 size) { in FloatToBFloat16() function
|
| D | numeric_types.h | 45 static inline tensorflow::bfloat16 FloatToBFloat16(float float_val) { in FloatToBFloat16() function
|
| /external/tensorflow/tensorflow/security/fuzzing/ |
| D | bfloat16_fuzz.cc | 40 tensorflow::FloatToBFloat16(float_originals, bfloats, array_size); in LLVMFuzzerTestOneInput()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 747 FloatToBFloat16(out_reshaped_float.flat<float>().data(),
|