diff options
| author |  Michael Case <mikecase@google.com> | 2018-09-25 18:52:12 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 18:56:30 -0700 |
| commit | d84159b6694773a558c8eeef63ef79b4034b8fa7 (patch) | |
| tree | 59695afb2098cbcc39fbb9b9ca667a891435a9bf /tensorflow/contrib/saved_model | |
| parent | c1e303ed8fa1bf11aaea16e68b14ba2f5ab5dde0 (diff) | |
Update BUILD files so that Estimator code depends on TF thru tf_no_contrib.
PiperOrigin-RevId: 214541221
Diffstat (limited to 'tensorflow/contrib/saved_model')
| -rw-r--r-- | tensorflow/contrib/saved_model/BUILD | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/contrib/saved_model/BUILD b/tensorflow/contrib/saved_model/BUILD index 4ca5274b2e..291ff83791 100644 --- a/tensorflow/contrib/saved_model/BUILD +++ b/tensorflow/contrib/saved_model/BUILD @@ -92,10 +92,7 @@ py_library( "//tensorflow/python:platform", "//tensorflow/python:saver", "//tensorflow/python:util", - "//tensorflow/python/estimator", - "//tensorflow/python/estimator:export", - "//tensorflow/python/estimator:keras", - "//tensorflow/python/estimator:model_fn", + "//tensorflow/python/estimator:estimator_py", "//tensorflow/python/keras:engine", "//tensorflow/python/saved_model", ], @@ -111,6 +108,7 @@ py_test( ":keras_saved_model", "//tensorflow/python:client_testlib", "//tensorflow/python:training", + "//tensorflow/python/estimator:estimator_py", "//tensorflow/python/keras", "//third_party/py/numpy", "@absl_py//absl/testing:parameterized", |