diff options
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r-- | tensorflow/contrib/training/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD index 627698e1da..d5c1b42f0c 100644 --- a/tensorflow/contrib/training/BUILD +++ b/tensorflow/contrib/training/BUILD @@ -280,6 +280,7 @@ py_test( "//tensorflow/python:platform", "//tensorflow/python:training", "//tensorflow/python:variables", + "//tensorflow/python/ops/losses", "//third_party/py/numpy", ], ) |