STT-tensorflow/tensorflow/python/keras/mixed_precision
A. Unique TensorFlower 034633f23b PY2 removal cleanup
PiperOrigin-RevId: 352106691
Change-Id: I382d53c64f0d29da430b8cb6d2395a2cb281509e
2021-01-15 16:48:57 -08:00
..
testdata Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
__init__.py Fix various links in Keras docstrings. 2020-04-15 19:59:22 -07:00
autocast_variable_test.py Fix issue when using mixed precision with RMSprop. 2020-12-14 20:18:29 -08:00
autocast_variable.py Do not use private convert_to_tensor API in autocast_variable.py. 2021-01-05 18:41:01 -08:00
BUILD PY2 removal cleanup 2021-01-15 16:48:57 -08:00
device_compatibility_check_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
device_compatibility_check.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
get_layer_policy_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
get_layer_policy.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
keras_test.py Small test fix: avoid creating optimizer objects inside the fn passed to strategy.run. This ends up creating separate optimizer objects for each replica in MirroredStrategy. 2020-11-30 11:58:07 -08:00
layer_correctness_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
loss_scale_benchmark.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
loss_scale_optimizer_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
loss_scale_optimizer.py Improve mixed precision docstrings. 2020-10-29 17:50:24 -07:00
loss_scale.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
mixed_precision_graph_rewrite_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
policy_test.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
policy.py merge from master 2020-10-30 15:04:50 +03:00
test_util.py Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00