Carlos Mocholí 6aeebf1bd3 Remove unnecessary dependency available checks (#10050)		2021-10-20 16:21:37 +00:00
..
__init__.py	Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026)	2021-10-19 21:09:37 +00:00
apex_amp.py	Isolate optimizer step logic to the `PrecisionPlugin` (#10029)	2021-10-20 15:43:08 +00:00
deepspeed_precision.py	Isolate optimizer step logic to the `PrecisionPlugin` (#10029)	2021-10-20 15:43:08 +00:00
double.py	Introduce `PrecisionPlugin.forward_context()` (#9988)	2021-10-18 12:58:19 +00:00
fully_sharded_native_amp.py	CI: precommit - docformatter (#8584)	2021-09-06 12:49:09 +00:00
ipu_precision.py	Isolate optimizer step logic to the `PrecisionPlugin` (#10029)	2021-10-20 15:43:08 +00:00
mixed.py	CI: precommit - docformatter (#8584)	2021-09-06 12:49:09 +00:00
native_amp.py	Remove unnecessary dependency available checks (#10050)	2021-10-20 16:21:37 +00:00
precision_plugin.py	Update backward hook for `PrecisionPlugin` (#10008)	2021-10-19 10:51:45 +00:00
sharded_native_amp.py	Remove `_NATIVE_AMP_AVAILABLE` checks (#9747)	2021-09-29 15:34:26 +02:00
tpu.py	Isolate optimizer step logic to the `PrecisionPlugin` (#10029)	2021-10-20 15:43:08 +00:00
tpu_bf16.py	Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026)	2021-10-19 21:09:37 +00:00