lightning/pytorch_lightning/accelerators
William Falcon 810b445097
ref: apex plugin (#3502)
* ref: apex plugin

* ref: apex plugin

* ref: apex plugin
2020-09-15 06:02:42 -04:00
..
__init__.py ref: merge backends x/n (#3478) 2020-09-12 13:55:55 -04:00
accelerator_connector.py ref: merge backends x/n (#3478) 2020-09-12 13:55:55 -04:00
base_backend.py ref: move specific accelerator code x/n (#3457) 2020-09-11 10:56:21 -04:00
cpu_backend.py ref: organize args 4/n (#3456) 2020-09-10 21:58:47 -04:00
ddp2_backend.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
ddp_backend.py ref: merge backends x/n (#3482) 2020-09-12 16:28:29 -04:00
ddp_base_backend.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
ddp_cpu_spawn_backend.py ref: merge backends x/n (#3482) 2020-09-12 16:28:29 -04:00
ddp_spawn_backend.py ref: merge backends x/n (#3480) 2020-09-12 15:27:11 -04:00
dp_backend.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
gpu_backend.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
horovod_backend.py ref: apex plugin (#3502) 2020-09-15 06:02:42 -04:00
tpu_backend.py ref: group connectors (#3472) 2020-09-11 23:33:09 -04:00