Carlos Mocholí d45897d522 Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026) ... Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2021-10-19 21:09:37 +00:00
..
__init__.py	Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026)	2021-10-19 21:09:37 +00:00
apex_amp.py	Run plugin closure before `on_before_optimizer_step` [1/2] (#9288)	2021-09-07 11:52:20 +00:00
deepspeed_precision.py	Update backward hook for `PrecisionPlugin` (#10008)	2021-10-19 10:51:45 +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	CI: precommit - docformatter (#8584)	2021-09-06 12:49:09 +00:00
mixed.py	CI: precommit - docformatter (#8584)	2021-09-06 12:49:09 +00:00
native_amp.py	Update backward hook for `PrecisionPlugin` (#10008)	2021-10-19 10:51:45 +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	Add `TPUPrecisionPlugin` (#10020)	2021-10-19 17:48:57 +00:00
tpu_bf16.py	Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026)	2021-10-19 21:09:37 +00:00