experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
mm-fix-macos-protobuf-segfault
STT-tensorflow/tensorflow/python/keras/mixed_precision
History
Reed Wanderman-Milne ab80014d88 Fix issue when using mixed precision with RMSprop. 
Before, accessing the `op` attribute on the return value of AutoCastVariable.assign in Eager mode would raise an AttributeError instead of returning None. Accessing the `op` attribute on an AutoCastVariable itself (that is not the return value of `assign`) still raises an AttributeError, to be consistent with tf.Variable.

Resolves https://github.com/tensorflow/tensorflow/issues/45536.

PiperOrigin-RevId: 347524886
Change-Id: I663731c0ff4c557608eae352096a527e4dcabb18
2020-12-14 20:18:29 -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 Fix issue when using mixed precision with RMSprop. 2020-12-14 20:18:29 -08:00
BUILD Some internal change 2020-11-13 15:28:21 -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