TPUBf16PrecisionPlugin¶
- class pytorch_lightning.plugins.precision.TPUBf16PrecisionPlugin[source]¶
Bases:
pytorch_lightning.plugins.precision.tpu.TPUPrecisionPlugin
Plugin that enables bfloats on TPUs.
Bases: pytorch_lightning.plugins.precision.tpu.TPUPrecisionPlugin
Plugin that enables bfloats on TPUs.