diff options
author | James Qin <jamesqin@google.com> | 2018-05-14 17:51:11 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-14 17:53:46 -0700 |
commit | 4c2cb712c7d7be93533da240d4c8e55e69d79625 (patch) | |
tree | 2ab6a3445c6df34d60106f9ee2a4f2c5184f1a4b /tensorflow/contrib/__init__.py | |
parent | e1a49f30435096c7e0817dde2e472c85db143a81 (diff) |
Introduce LossScalingOptimizer for mixed precision training.
PiperOrigin-RevId: 196597196
Diffstat (limited to 'tensorflow/contrib/__init__.py')
-rw-r--r-- | tensorflow/contrib/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/__init__.py b/tensorflow/contrib/__init__.py index 9f5459f41d..9aad772f0a 100644 --- a/tensorflow/contrib/__init__.py +++ b/tensorflow/contrib/__init__.py @@ -60,6 +60,7 @@ from tensorflow.contrib import lookup from tensorflow.contrib import losses from tensorflow.contrib import memory_stats from tensorflow.contrib import metrics +from tensorflow.contrib import mixed_precision from tensorflow.contrib import model_pruning from tensorflow.contrib import nccl from tensorflow.contrib import nn |