diff options
| author |  Petros Mol <pmol@google.com> | 2017-04-03 14:18:28 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-03 15:24:55 -0700 |
| commit | 01f03299baa4cb1f1b59ee7e370dc938aaba7a20 (patch) | |
| tree | 2d7cb7ad5276c4abbdcb2cfbafbfb4fddb0f1fd6 /tensorflow/contrib/kernel_methods | |
| parent | e18c7a5332b97ddcaad7f5485c46ad8dc8fcd274 (diff) | |
Add kernel_methods package to contrib
Change: 152064888
Diffstat (limited to 'tensorflow/contrib/kernel_methods')
| -rw-r--r-- | tensorflow/contrib/kernel_methods/BUILD | 22 | ||||
| -rw-r--r-- | tensorflow/contrib/kernel_methods/__init__.py | 1 |
2 files changed, 15 insertions, 8 deletions
diff --git a/tensorflow/contrib/kernel_methods/BUILD b/tensorflow/contrib/kernel_methods/BUILD index b37cbc119f..fccaa3abd4 100644 --- a/tensorflow/contrib/kernel_methods/BUILD +++ b/tensorflow/contrib/kernel_methods/BUILD @@ -21,9 +21,14 @@ py_library( ":dense_kernel_mapper_py", "//tensorflow/contrib/layers:layers_py", "//tensorflow/contrib/learn", - "//tensorflow/python:framework", - "//tensorflow/python:ops", + "//tensorflow/python:array_ops", + "//tensorflow/python:constant_op", + "//tensorflow/python:dtypes", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform", + "//tensorflow/python:util", "//third_party/py/numpy", + "@six_archive//:six", ], ) @@ -31,6 +36,7 @@ py_library( name = "dense_kernel_mapper_py", srcs = ["python/mappers/dense_kernel_mapper.py"], srcs_version = "PY2AND3", + deps = ["@six_archive//:six"], ) py_test( @@ -40,12 +46,12 @@ py_test( deps = [ ":dense_kernel_mapper_py", ":kernel_methods", - "//tensorflow/python:client_testlib", - "//tensorflow/python:framework", "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:framework_test_lib", + "//tensorflow/python:math_ops", "//tensorflow/python:nn", - "//tensorflow/python:ops", + "//tensorflow/python:platform_test", + "//tensorflow/python:random_ops", ], ) @@ -55,10 +61,12 @@ py_test( srcs_version = "PY2AND3", deps = [ ":kernel_methods", - "//tensorflow/python:client_testlib", + "//tensorflow/contrib/layers:layers_py", + "//tensorflow/contrib/learn", "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:framework_test_lib", - "//tensorflow/python:ops", + "//tensorflow/python:platform_test", + "//tensorflow/python:sparse_tensor", "//third_party/py/numpy", ], ) diff --git a/tensorflow/contrib/kernel_methods/__init__.py b/tensorflow/contrib/kernel_methods/__init__.py index 1a3a0ab77a..a715df1bf1 100644 --- a/tensorflow/contrib/kernel_methods/__init__.py +++ b/tensorflow/contrib/kernel_methods/__init__.py @@ -22,7 +22,6 @@ from __future__ import division from __future__ import print_function from tensorflow.contrib.kernel_methods.python.kernel_estimators import KernelLinearClassifier -from tensorflow.contrib.kernel_methods.python.mappers import dense_kernel_mapper from tensorflow.contrib.kernel_methods.python.mappers.random_fourier_features import RandomFourierFeatureMapper from tensorflow.python.util.all_util import remove_undocumented |