diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r-- | tensorflow/core/kernels/BUILD | 100 |
1 files changed, 88 insertions, 12 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index d3710a4b5c..2cb54bd973 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -368,6 +368,7 @@ cc_library( cc_library( name = "queue_op", + srcs = ["queue_op.cc"], hdrs = ["queue_op.h"], deps = [ ":queue_base", @@ -881,7 +882,6 @@ tf_kernel_library( "tile_functor_gpu.cu.cc", ], prefix = "tile_ops", - textual_hdrs = ["tile_ops_gpu_impl.h"], deps = ARRAY_DEPS, ) @@ -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", @@ -1885,9 +1908,10 @@ cc_library( name = "fifo_queue", srcs = ["fifo_queue.cc"], hdrs = ["fifo_queue.h"], - visibility = ["//visibility:private"], + visibility = [":friends"], deps = [ ":queue_base", + ":queue_op", ":typed_queue", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -2085,6 +2109,7 @@ IMAGE_DEPS = [ "//tensorflow/core:jpeg_internal", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", + "//tensorflow/core:png_internal", "//tensorflow/core:protos_all_cc", ] @@ -2659,7 +2684,7 @@ tf_kernel_library( tf_kernel_library( name = "summary_image_op", prefix = "summary_image_op", - deps = LOGGING_DEPS, + deps = LOGGING_DEPS + ["//tensorflow/core:png_internal"], ) tf_kernel_library( @@ -2704,17 +2729,16 @@ cc_library( ], ) -MANIP_DEPS = [ - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:manip_ops_op_lib", - "//third_party/eigen3", -] - tf_kernel_library( name = "roll_op", prefix = "roll_op", - deps = MANIP_DEPS, + deps = [ + ":bounds_check", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:manip_ops_op_lib", + "//third_party/eigen3", + ], ) tf_cc_test( @@ -3383,6 +3407,14 @@ cc_library( ], ) +# Kernels for the nodes intented to be added to the graph by the Grappler optimizers. +cc_library( + name = "grappler", + deps = [ + ":unary_ops_composition", + ], +) + NN_DEPS = [ ":bounds_check", ":conv_2d", @@ -3919,6 +3951,8 @@ tf_cc_test( cc_library( name = "sparse", deps = [ + ":deserialize_sparse_string_op", + ":deserialize_sparse_variant_op", ":serialize_sparse_op", ":sparse_add_grad_op", ":sparse_add_op", @@ -4073,6 +4107,23 @@ tf_kernel_library( ) tf_kernel_library( + name = "deserialize_sparse_string_op", + prefix = "deserialize_sparse_string_op", + deps = SPARSE_DEPS + [ + ":reshape_util", + "//tensorflow/core:protos_all_cc", + ], +) + +tf_kernel_library( + name = "deserialize_sparse_variant_op", + prefix = "deserialize_sparse_variant_op", + deps = SPARSE_DEPS + [ + "//tensorflow/core:protos_all_cc", + ], +) + +tf_kernel_library( name = "sparse_tensors_map_ops", prefix = "sparse_tensors_map_ops", deps = SPARSE_DEPS, @@ -4798,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", @@ -5083,6 +5136,7 @@ filegroup( "padding_fifo_queue.cc", "padding_fifo_queue_op.cc", "queue_base.cc", + "queue_op.cc", "queue_ops.cc", "random_op.cc", "reduction_ops_all.cc", @@ -5171,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. @@ -5232,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( @@ -6259,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", ], ) @@ -6271,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", ], ) |