Adrian Wälchli 7fe8756917 [TPU] Proper half-precision implementation for XLA (#18213) ... Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>		2023-08-11 11:37:41 -04:00
..
conf	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
__init__.py	…
test_amp.py	ruff: enable & fixing RET (#17540)	2023-05-05 09:34:40 +00:00
test_cpu.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_ddp_fork_amp.py	Introduce new precision layout in PL (#16783)	2023-02-17 17:58:14 +01:00
test_fabric_integration.py	Enable `Fabric.call` to call hooks on the LightningModule (#17874)	2023-06-21 13:25:05 +00:00
test_gpu.py	Cast input before moving to device for all strategies (#18264)	2023-08-09 17:42:55 +02:00
test_hooks.py	Add `Trainer.init_module` and `LightningModule.configure_model` (#18004)	2023-07-14 19:15:05 +02:00
test_hparams.py	ruff: PT some more fixes (#17569)	2023-05-05 08:25:15 +02:00
test_onnx.py	ruff: autofix PT (#17541)	2023-05-04 11:50:39 -04:00
test_restore.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_torchscript.py	[pre-commit.ci] pre-commit suggestions (#17271)	2023-04-26 21:37:41 +02:00
test_tpu.py	[TPU] Proper half-precision implementation for XLA (#18213)	2023-08-11 11:37:41 -04:00