diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r-- | tensorflow/core/kernels/BUILD | 31 |
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"], |