diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 59 |
1 files changed, 27 insertions, 32 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 4169e842da..a5c62fef17 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -917,6 +917,25 @@ tf_cc_test( ) tf_cuda_cc_test( + name = "bincount_op_test", + size = "small", + srcs = ["bincount_op_test.cc"], + deps = [ + ":bincount_op", + ":ops_testutil", + ":ops_util", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:math_ops_op_lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + +tf_cuda_cc_test( name = "constant_op_test", size = "small", srcs = ["constant_op_test.cc"], @@ -1601,7 +1620,10 @@ DYNAMIC_DEPS = [ tf_kernel_library( name = "dynamic_partition_op", prefix = "dynamic_partition_op", - deps = DYNAMIC_DEPS, + deps = DYNAMIC_DEPS + [ + ":fill_functor", + ":gather_functor", + ] + if_cuda(["@cub_archive//:cub"]), ) tf_kernel_library( @@ -1671,7 +1693,7 @@ tf_kernel_library( ], ) -tf_cc_tests( +tf_cuda_cc_tests( name = "dynamic_op_test", size = "small", srcs = [ @@ -2554,8 +2576,9 @@ tf_kernel_library( tf_kernel_library( name = "bucketize_op", + gpu_srcs = ["cuda_device_array.h"], prefix = "bucketize_op", - deps = MATH_DEPS, + deps = ARRAY_DEPS, ) tf_kernel_library( @@ -3156,7 +3179,7 @@ tf_kernel_library( "//tensorflow/core:lib", "//tensorflow/core:lib_internal", "//third_party/eigen3", - ], + ] + if_cuda(["@cub_archive//:cub"]), ) tf_kernel_library( @@ -4420,15 +4443,6 @@ filegroup( "fill_functor.h", "function_ops.cc", "gather_functor.h", - "gather_nd_op.cc", - "gather_nd_op.h", - "gather_nd_op_cpu_impl.h", - "gather_nd_op_cpu_impl_0.cc", - "gather_nd_op_cpu_impl_1.cc", - "gather_nd_op_cpu_impl_2.cc", - "gather_nd_op_cpu_impl_3.cc", - "gather_nd_op_cpu_impl_4.cc", - "gather_nd_op_cpu_impl_5.cc", "gather_op.cc", "identity_n_op.cc", "identity_n_op.h", @@ -4522,10 +4536,6 @@ filegroup( "fused_batch_norm_op.h", "gemm_functors.h", "image_resizer_state.h", - "initializable_lookup_table.h", - "lookup_table_init_op.h", - "lookup_table_op.h", - "lookup_util.h", "maxpooling_op.h", "mfcc.h", "mfcc_dct.h", @@ -4542,7 +4552,6 @@ filegroup( "resize_nearest_neighbor_op.h", "reverse_op.h", "save_restore_tensor.h", - "segment_reduction_ops.h", "softplus_op.h", "softsign_op.h", "spacetobatch_functor.h", @@ -4592,8 +4601,6 @@ filegroup( "cwise_op_div.cc", "cwise_op_equal_to_1.cc", "cwise_op_equal_to_2.cc", - "cwise_op_not_equal_to_1.cc", - "cwise_op_not_equal_to_2.cc", "cwise_op_exp.cc", "cwise_op_floor.cc", "cwise_op_floor_div.cc", @@ -4635,7 +4642,6 @@ filegroup( "encode_wav_op.cc", "fake_quant_ops.cc", "fifo_queue.cc", - "fifo_queue_op.cc", "fused_batch_norm_op.cc", "population_count_op.cc", "population_count_op.h", @@ -4659,11 +4665,7 @@ filegroup( "depthtospace_op.cc", "dynamic_stitch_op.cc", "in_topk_op.cc", - "initializable_lookup_table.cc", "logging_ops.cc", - "lookup_table_init_op.cc", - "lookup_table_op.cc", - "lookup_util.cc", "lrn_op.cc", "maxpooling_op.cc", "mfcc.cc", @@ -4698,15 +4700,12 @@ filegroup( "save_op.cc", "save_restore_tensor.cc", "save_restore_v2_ops.cc", - "segment_reduction_ops.cc", "session_ops.cc", "softplus_op.cc", "softsign_op.cc", "spacetobatch_functor.cc", "spacetobatch_op.cc", "spacetodepth_op.cc", - "sparse_fill_empty_rows_op.cc", - "sparse_reshape_op.cc", "sparse_to_dense_op.cc", "spectrogram.cc", "spectrogram_op.cc", @@ -4729,7 +4728,6 @@ filegroup( "training_ops.cc", "transpose_functor_cpu.cc", "transpose_op.cc", - "unique_op.cc", "warn_about_ints.cc", "where_op.cc", "xent_op.cc", @@ -6243,11 +6241,8 @@ tf_kernel_library( srcs = ["summary_kernels.cc"], deps = [ ":summary_interface", - "//tensorflow/contrib/tensorboard/db:summary_db_writer", "//tensorflow/core:framework", - "//tensorflow/core:lib", "//tensorflow/core:summary_ops_op_lib", - "//tensorflow/core/lib/db:sqlite", ], ) |