diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-10 20:07:21 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-10 20:11:48 -0800 |
| commit | e0249a0589cda4fec00a88b7b3e5de497c08d3ea (patch) | |
| tree | fa474dcfbab0af918a5133a24b4ca6e7d9499f8c /tensorflow/contrib/model_pruning | |
| parent | 9c6eebabc71906d240338adc89fa838bd5635aa0 (diff) | |
Further BUILD cleanup in contrib/...
PiperOrigin-RevId: 175370768
Diffstat (limited to 'tensorflow/contrib/model_pruning')
| -rw-r--r-- | tensorflow/contrib/model_pruning/examples/cifar10/BUILD | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/model_pruning/examples/cifar10/BUILD b/tensorflow/contrib/model_pruning/examples/cifar10/BUILD index 299278ae75..e7848adcc5 100644 --- a/tensorflow/contrib/model_pruning/examples/cifar10/BUILD +++ b/tensorflow/contrib/model_pruning/examples/cifar10/BUILD @@ -39,6 +39,7 @@ py_library( deps = [ ":cifar10_input", "//tensorflow:tensorflow_py", + "//tensorflow/contrib/model_pruning:pruning", ], ) @@ -50,6 +51,8 @@ py_binary( srcs_version = "PY2AND3", deps = [ ":cifar10_pruning", + "//tensorflow:tensorflow_py", + "//third_party/py/numpy", ], ) @@ -61,6 +64,8 @@ py_binary( srcs_version = "PY2AND3", deps = [ ":cifar10_pruning", + "//tensorflow:tensorflow_py", + "//tensorflow/contrib/model_pruning:pruning", ], ) |