diff options
Diffstat (limited to 'tensorflow/python/feature_column/BUILD')
-rw-r--r-- | tensorflow/python/feature_column/BUILD | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/tensorflow/python/feature_column/BUILD b/tensorflow/python/feature_column/BUILD index 295d4ca094..80707030e6 100644 --- a/tensorflow/python/feature_column/BUILD +++ b/tensorflow/python/feature_column/BUILD @@ -48,6 +48,39 @@ py_library( ], ) +py_library( + name = "feature_column_v2", + srcs = ["feature_column_v2.py"], + srcs_version = "PY2AND3", + deps = [ + "//tensorflow/python:array_ops", + "//tensorflow/python:check_ops", + "//tensorflow/python:control_flow_ops", + "//tensorflow/python:dtypes", + "//tensorflow/python:embedding_ops", + "//tensorflow/python:framework_ops", + "//tensorflow/python:init_ops", + "//tensorflow/python:lookup_ops", + "//tensorflow/python:math_ops", + "//tensorflow/python:nn_ops", + "//tensorflow/python:parsing_ops", + "//tensorflow/python:platform", + "//tensorflow/python:resource_variable_ops", + "//tensorflow/python:sparse_ops", + "//tensorflow/python:sparse_tensor", + "//tensorflow/python:string_ops", + "//tensorflow/python:template", + "//tensorflow/python:tensor_shape", + "//tensorflow/python:training", + "//tensorflow/python:util", + "//tensorflow/python:variable_scope", + "//tensorflow/python:variables", + "//tensorflow/python/keras", + "//third_party/py/numpy", + "@six_archive//:six", + ], +) + filegroup( name = "vocabulary_testdata", srcs = [ @@ -92,3 +125,38 @@ py_test( "//tensorflow/python/estimator:numpy_io", ], ) + +py_test( + name = "feature_column_v2_test", + srcs = ["feature_column_v2_test.py"], + data = [":vocabulary_testdata"], + srcs_version = "PY2AND3", + tags = [ + "no_cuda_on_cpu_tap", + "no_pip", + ], + deps = [ + ":feature_column_py", + ":feature_column_v2", + "//tensorflow/core:protos_all_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:constant_op", + "//tensorflow/python:dtypes", + "//tensorflow/python:errors", + "//tensorflow/python:framework_ops", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:lookup_ops", + "//tensorflow/python:parsing_ops", + "//tensorflow/python:partitioned_variables", + "//tensorflow/python:session", + "//tensorflow/python:sparse_tensor", + "//tensorflow/python:training", + "//tensorflow/python:variable_scope", + "//tensorflow/python:variables", + "//tensorflow/python/eager:backprop", + "//tensorflow/python/eager:context", + "//tensorflow/python/estimator:numpy_io", + "//third_party/py/numpy", + ], +) |