diff options
author | 2018-05-14 17:51:11 -0700 | |
---|---|---|
committer | 2018-05-14 17:53:46 -0700 | |
commit | 4c2cb712c7d7be93533da240d4c8e55e69d79625 (patch) | |
tree | 2ab6a3445c6df34d60106f9ee2a4f2c5184f1a4b /tensorflow/contrib/BUILD | |
parent | e1a49f30435096c7e0817dde2e472c85db143a81 (diff) |
Introduce LossScalingOptimizer for mixed precision training.
PiperOrigin-RevId: 196597196
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index abdbdb4cd2..0f9c80404a 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -71,6 +71,7 @@ py_library( "//tensorflow/contrib/memory_stats:memory_stats_py", "//tensorflow/contrib/meta_graph_transform", "//tensorflow/contrib/metrics:metrics_py", + "//tensorflow/contrib/mixed_precision:mixed_precision", "//tensorflow/contrib/model_pruning", "//tensorflow/contrib/nccl:nccl_py", "//tensorflow/contrib/nearest_neighbor:nearest_neighbor_py", |