diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r-- | tensorflow/core/kernels/BUILD | 45 |
1 files changed, 34 insertions, 11 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index a7c7551ea6..7ca186eaec 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -28,16 +28,6 @@ cc_library( ], ) -cc_library( - name = "bounds_check", - hdrs = ["bounds_check.h"], - visibility = ["//visibility:private"], - deps = [ - "//tensorflow/core:framework", - "//third_party/eigen3", - ], -) - tf_kernel_library( name = "concat_lib", srcs = ["concat_lib_cpu.cc"], @@ -153,7 +143,6 @@ tf_proto_library( cc_api_version = 2, go_api_version = 2, java_api_version = 2, - py_api_version = 2, ) cc_library( @@ -200,6 +189,18 @@ cc_library( ], ) +# Private support libraries --------------------------------------------------- + +cc_library( + name = "bounds_check", + hdrs = ["bounds_check.h"], + visibility = ["//visibility:private"], + deps = [ + "//tensorflow/core:framework", + "//third_party/eigen3", + ], +) + # OpKernel libraries ---------------------------------------------------------- tf_kernel_libraries( @@ -652,6 +653,7 @@ tf_kernel_libraries( "sparse_matmul_op", ], deps = [ + ":bounds_check", ":fill_functor", ":transpose_functor", "//tensorflow/core:core_cpu", @@ -734,6 +736,7 @@ tf_kernel_libraries( "xent_op", ], deps = [ + ":bounds_check", ":conv_2d", ":conv_ops", ":depthwise_conv_op", @@ -980,6 +983,8 @@ filegroup( ], ) +# Core kernels we want on Android. Only a subset of kernels to keep +# base library small. filegroup( name = "android_core_ops", srcs = [ @@ -1036,6 +1041,22 @@ filegroup( ], ) +# Other kernels we may want on Android. +# +# The kernels can be consumed as a whole or in two groups for +# supporting separate compilation. Note that the split into groups +# is entirely for improving compilation time, and not for +# organizational reasons; you should not depend on any +# of those groups independently. +filegroup( + name = "android_extended_ops", + srcs = [ + ":android_extended_ops_group1", + ":android_extended_ops_group2", + ], + visibility = ["//visibility:public"], +) + filegroup( name = "android_extended_ops_headers", srcs = [ @@ -1090,6 +1111,7 @@ filegroup( "cwise_op_sub.cc", "cwise_op_tanh.cc", "dynamic_partition_op.cc", + ":android_extended_ops_headers", ], ) @@ -1122,6 +1144,7 @@ filegroup( "transpose_op.cc", "where_op.cc", "xent_op.cc", + ":android_extended_ops_headers", ], ) |