From 2d759d55a6873ad4d551a597857f1ab2d9cdce43 Mon Sep 17 00:00:00 2001 From: Nathan Silberman Date: Wed, 22 Feb 2017 10:03:04 -0800 Subject: Speeding up and refactoring contrib/training_test.py Change: 148232625 --- tensorflow/contrib/training/BUILD | 1 + 1 file changed, 1 insertion(+) (limited to 'tensorflow/contrib/training/BUILD') 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", ], ) -- cgit v1.2.3