diff options
| author |  Brennan Saeta <saeta@google.com> | 2018-07-04 19:01:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-04 19:04:10 -0700 |
| commit | 9103612c1d30a9e77bd1af83a8bc036dcee84b60 (patch) | |
| tree | 046e632714c946246676212be691c9d44b63668c /tensorflow/contrib/bigtable | |
| parent | eb8a110e52a8e8ed7b9db48e01115d3110e918c2 (diff) | |
[tf.data / Bigtable] Refactor BUILD file
PiperOrigin-RevId: 203319567
Diffstat (limited to 'tensorflow/contrib/bigtable')
| -rw-r--r-- | tensorflow/contrib/bigtable/BUILD | 23 |
1 files changed, 10 insertions, 13 deletions
diff --git a/tensorflow/contrib/bigtable/BUILD b/tensorflow/contrib/bigtable/BUILD index 5c15d21e35..037645aa6e 100644 --- a/tensorflow/contrib/bigtable/BUILD +++ b/tensorflow/contrib/bigtable/BUILD @@ -39,14 +39,17 @@ tf_custom_op_py_library( ], ) +KERNEL_FILES = [ + "kernels/bigtable_kernels.cc", + "kernels/bigtable_lookup_dataset_op.cc", + "kernels/bigtable_prefix_key_dataset_op.cc", + "kernels/bigtable_range_key_dataset_op.cc", + "kernels/bigtable_scan_dataset_op.cc", +] + tf_custom_op_library( name = "python/ops/_bigtable.so", - srcs = [ - "kernels/bigtable_kernels.cc", - "kernels/bigtable_lookup_dataset_op.cc", - "kernels/bigtable_prefix_key_dataset_op.cc", - "kernels/bigtable_range_key_dataset_op.cc", - "kernels/bigtable_scan_dataset_op.cc", + srcs = KERNEL_FILES + [ "ops/bigtable_ops.cc", ], deps = [ @@ -69,13 +72,7 @@ tf_gen_op_libs( tf_kernel_library( name = "bigtable_kernels", - srcs = [ - "kernels/bigtable_kernels.cc", - "kernels/bigtable_lookup_dataset_op.cc", - "kernels/bigtable_prefix_key_dataset_op.cc", - "kernels/bigtable_range_key_dataset_op.cc", - "kernels/bigtable_scan_dataset_op.cc", - ], + srcs = KERNEL_FILES, deps = [ ":bigtable_lib_cc", "//tensorflow/core:framework_headers_lib", |