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