aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/image_retraining
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-21 15:05:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-21 15:10:20 -0700
commit0a379aefe4f76966fe3cca0bb91e1c5406019451 (patch)
tree804cdab0e295bea994d35c0ad7e5981656be1c6f /tensorflow/examples/image_retraining
parent12c4775b378fefe1351c454d9516775663559c4f (diff)
Further BUILD cleanup
PiperOrigin-RevId: 159749869
Diffstat (limited to 'tensorflow/examples/image_retraining')
-rw-r--r--tensorflow/examples/image_retraining/BUILD7
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/examples/image_retraining/BUILD b/tensorflow/examples/image_retraining/BUILD
index 394c413b33..c8c136ac14 100644
--- a/tensorflow/examples/image_retraining/BUILD
+++ b/tensorflow/examples/image_retraining/BUILD
@@ -18,6 +18,7 @@ py_binary(
"//tensorflow:tensorflow_py",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
+ "//tensorflow/python:graph_util",
"//tensorflow/python:platform",
"//tensorflow/python:util",
"//third_party/py/numpy",
@@ -50,10 +51,16 @@ py_test(
],
srcs_version = "PY2AND3",
deps = [
+ ":label_image",
":retrain",
"//tensorflow:tensorflow_py",
"//tensorflow/python:framework_test_lib",
+ "//tensorflow/python:graph_util",
+ "//tensorflow/python:platform",
"//tensorflow/python:platform_test",
+ "//tensorflow/python:tensor_shape",
+ "//tensorflow/python:util",
+ "//third_party/py/numpy",
],
)