lightning/pytorch_lightning/accelerators
Carlos Mocholí e8beceb631
Add `TPUPrecisionPlugin` (#10020)
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-10-19 17:48:57 +00:00
..
__init__.py CI: yesqa (#8564) 2021-08-02 16:05:56 +00:00
accelerator.py [2/n] Directly call TrainingTypePlugin APIs instead of going through the Accelerator (#9901) 2021-10-14 17:38:22 +02:00
cpu.py [2/4] Add DeviceStatsMonitor callback (#9712) 2021-10-13 18:29:36 +00:00
gpu.py Avoid deprecation warning after #9901 (#9951) 2021-10-16 17:36:25 +01:00
ipu.py [2/4] Add DeviceStatsMonitor callback (#9712) 2021-10-13 18:29:36 +00:00
tpu.py Add `TPUPrecisionPlugin` (#10020) 2021-10-19 17:48:57 +00:00