aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/BUILD
diff options
context:
space:
mode:
authorGravatar James Qin <jamesqin@google.com>2018-05-14 17:51:11 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-14 17:53:46 -0700
commit4c2cb712c7d7be93533da240d4c8e55e69d79625 (patch)
tree2ab6a3445c6df34d60106f9ee2a4f2c5184f1a4b /tensorflow/contrib/BUILD
parente1a49f30435096c7e0817dde2e472c85db143a81 (diff)
Introduce LossScalingOptimizer for mixed precision training.
PiperOrigin-RevId: 196597196
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r--tensorflow/contrib/BUILD1
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",