Kaushik B 94fcaaf5d7 Add `debug` flag to TPU Training Plugins (PT_XLA_DEBUG) (#7219)		2021-04-27 20:34:25 +00:00
..
__init__.py	PoC: Accelerator refactor (#5743)	2021-02-12 15:48:56 -05:00
accelerator.py	Add back `clip_gradients(model)` (#7231)	2021-04-27 11:34:02 +00:00
cpu.py	Accelerator API docs (#6936)	2021-04-10 12:25:07 +05:30
gpu.py	Properly set `LightningModule.device` after model replacement (#7188)	2021-04-23 16:36:52 +02:00
tpu.py	Add `debug` flag to TPU Training Plugins (PT_XLA_DEBUG) (#7219)	2021-04-27 20:34:25 +00:00