diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r-- | tensorflow/core/kernels/BUILD | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index fffcb980db..a493452777 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -702,6 +702,39 @@ tf_kernel_library( ) tf_kernel_library( + name = "compare_and_bitpack_op", + srcs = ["compare_and_bitpack_op.cc"], + hdrs = ["compare_and_bitpack_op.h"], + gpu_srcs = [ + "compare_and_bitpack_op.h", + "compare_and_bitpack_op_gpu.cu.cc", + ], + deps = ARRAY_DEPS, +) + +# TODO(ebrevdo): Add benchmarks once the op is in the autogen array namespace. +# tf_cuda_cc_test( +# name = "compare_and_bitpack_op_test", +# srcs = ["compare_and_bitpack_op_test.cc"], +# deps = [ +# ":array", +# ":ops_testutil", +# ":ops_util", +# "//third_party/eigen3", +# "//tensorflow/cc:cc_ops", +# "//tensorflow/cc:cc_ops_internal", +# "//tensorflow/core:core_cpu", +# "//tensorflow/core:core_cpu_internal", +# "//tensorflow/core:framework", +# "//tensorflow/core:lib", +# "//tensorflow/core:protos_all_cc", +# "//tensorflow/core:test", +# "//tensorflow/core:test_main", +# "//tensorflow/core:testlib", +# ], +# ) + +tf_kernel_library( name = "reshape_op", prefix = "reshape_op", deps = ARRAY_DEPS, @@ -2344,10 +2377,12 @@ cc_library( ":bucketize_op", ":cast_op", ":check_numerics_op", + ":compare_and_bitpack_op", ":cross_op", ":cwise_op", ":fft_ops", ":matmul_op", + ":population_count_op", ":reduction_ops", ":scan_ops", ":segment_reduction_ops", @@ -2410,6 +2445,12 @@ tf_kernel_library( ) tf_kernel_library( + name = "population_count_op", + prefix = "population_count_op", + deps = MATH_DEPS, +) + +tf_kernel_library( name = "fft_ops", prefix = "fft_ops", deps = MATH_DEPS + [ @@ -4292,6 +4333,8 @@ filegroup( "fake_quant_ops.cc", "fifo_queue.cc", "fused_batch_norm_op.cc", + "population_count_op.cc", + "population_count_op.h", "winograd_transform.h", ":android_extended_ops_headers", ] + select({ |