Carlos Mocholí 8822117200 Return the output of the optimizer step (#11711) ... Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>		2022-02-09 09:37:13 +00:00
..
__init__.py	Remove the "_precision" suffix from some precision plugin files (#10052)	2021-11-19 17:37:39 +00:00
apex_amp.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
deepspeed.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
double.py	Introduce `PrecisionPlugin.forward_context()` (#9988)	2021-10-18 12:58:19 +00:00
fully_sharded_native_amp.py	Fix deadlinks in docs (#10739)	2021-12-02 09:05:32 +00:00
ipu.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
mixed.py	CI: precommit - docformatter (#8584)	2021-09-06 12:49:09 +00:00
native_amp.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
precision_plugin.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
sharded_native_amp.py	Pass the scaler as an input to `NativeMixedPrecisionPlugin` (#10055)	2021-10-28 14:13:53 +00:00
tpu.py	Return the output of the optimizer step (#11711)	2022-02-09 09:37:13 +00:00
tpu_bf16.py	Fix BF16 teardown for TPU precision plugin (#10990)	2021-12-22 03:47:14 +00:00