diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 49 |
1 files changed, 48 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 7599cf7db2..2cb54bd973 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -1105,6 +1105,29 @@ tf_cc_test( ], ) +tf_cuda_cc_test( + name = "depthwise_conv_ops_test", + size = "small", + srcs = ["depthwise_conv_ops_test.cc"], + tags = ["requires-gpu-sm35"], + deps = [ + ":conv_ops", + ":image", + ":ops_testutil", + ":ops_util", + "//tensorflow/cc:cc_ops", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:framework_internal", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:tensorflow", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + tf_cc_test( name = "decode_wav_op_test", size = "small", @@ -4826,6 +4849,8 @@ filegroup( "cast_op_impl_int64.cc", "cast_op_impl_int8.cc", "cast_op_impl_uint16.cc", + "cast_op_impl_uint32.cc", + "cast_op_impl_uint64.cc", "cast_op_impl_uint8.cc", "concat_lib.h", "concat_lib_cpu.cc", @@ -5200,6 +5225,16 @@ filegroup( visibility = ["//visibility:public"], ) +ANDROID_TEXTUAL_HDRS = [ + "gather_nd_op_cpu_impl.h", + "gemm_functors.h", + "mirror_pad_op_cpu_impl.h", + "scatter_nd_op_cpu_impl.h", + "slice_op_cpu_impl.h", + "strided_slice_op_impl.h", + "tile_ops_cpu_impl.h", +] + # A file group which contains nearly all available operators which # may work on Android. This is intended to be used with selective # registration. @@ -5261,10 +5296,20 @@ filegroup( "batch_kernels.*", "regex_full_match_op.cc", "regex_replace_op.cc", - ], + # Ops that are inherently incompatible with Android (e.g. tied to x86 platform). + "mkl_*", + "xsmm_*", + "cwise_ops_sycl_common.h", + ] + ANDROID_TEXTUAL_HDRS, ), visibility = ["//visibility:public"], ) + +filegroup( + name = "android_all_ops_textual_hdrs", + srcs = ANDROID_TEXTUAL_HDRS, + visibility = ["//visibility:public"], +) # LINT.ThenChange(//tensorflow/contrib/makefile/tf_op_files.txt) cc_library( @@ -6288,6 +6333,7 @@ tf_kernel_library( "//tensorflow/core:lib", "//tensorflow/core/util/proto:decode", "//tensorflow/core/util/proto:descriptors", + "//tensorflow/core/util/proto:proto_utils", "//third_party/eigen3", ], ) @@ -6300,6 +6346,7 @@ tf_kernel_library( "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core/util/proto:descriptors", + "//tensorflow/core/util/proto:proto_utils", "//third_party/eigen3", ], ) |