aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r--tensorflow/core/kernels/BUILD31
1 files changed, 31 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
index 3a920f26f3..1ca9c7b7f5 100644
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@@ -958,6 +958,37 @@ tf_kernel_library(
]) + ARRAY_DEPS,
)
+cc_library(
+ name = "ragged_ops",
+ deps = [
+ ":ragged_gather_op",
+ ],
+)
+
+tf_kernel_library(
+ name = "ragged_gather_op",
+ srcs = ["ragged_gather_op.cc"],
+ deps = [
+ "//tensorflow/core:framework",
+ "//tensorflow/core:ragged_array_ops_op_lib",
+ ],
+)
+
+tf_cc_test(
+ name = "ragged_gather_op_test",
+ size = "small",
+ srcs = ["ragged_gather_op_test.cc"],
+ deps = [
+ ":ragged_gather_op",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:ragged_array_ops_op_lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ "//tensorflow/core:testlib",
+ "//tensorflow/core/kernels:ops_testutil",
+ ],
+)
+
tf_kernel_library(
name = "cudnn_rnn_kernels",
srcs = ["cudnn_rnn_ops.cc"],