Scott Zhu 710cc7e0e5 Fork the keras related MP test to keras folder. ... PiperOrigin-RevId: 316183309 Change-Id: I89aca251b05817811a382e098cf0e882714dbd45		2020-06-12 15:12:04 -07:00
..
loss_scale_optimizer_test.py	Fork the mp test util function to break the dependency between Keras and TF.	2020-06-11 12:43:30 -07:00
loss_scale_optimizer.py	Merge pull request #37965 from zhuzilin:keras-amp-variables-dev	2020-04-06 12:54:33 -07:00
loss_scale_test.py	Implement __repr__ for LossScale subclasses.	2019-08-09 12:08:28 -07:00
loss_scale.py	remove the added api on loss_scale	2020-03-30 09:25:43 +08:00
loss_scaling_gradient_tape_test.py	Update loss_scaling_gradient_tape_test to not rely on Keras.	2020-06-11 16:36:18 -07:00
loss_scaling_gradient_tape.py	Apply name change(experimental_run_v2 -> run) for all callers in Tensorflow.	2020-03-18 10:35:05 -07:00
mixed_precision_global_state.py	Raise mixed precision error in fewer cases.	2020-03-13 14:33:05 -07:00
mixed_precision_test.py	Fork the keras related MP test to keras folder.	2020-06-12 15:12:04 -07:00
mixed_precision.py	Try to break the dependency of MP in training and keras.	2020-03-25 21:20:10 -07:00