How do I know if my hardware supports bf16?

TensorFlow
,
1

I want to try to train and use a tflite model, but it seems that not all hardware supports bf16. How can I judge whether the hardware supports bf16?

3

There are bfloat16 instructions checks for CPU in:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/platform/cpu_info.cc

Are you interested also for GPU?

1 Like