* Fix tests of mixed precision now that experimental is deprecated * Fix mixed precision in training_args_tf.py too