From 0c6baae5af46bb22ea52db724e2194845d3bbf8c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 9 Oct 2018 12:24:05 -0700 Subject: Add RaggedTensors to tf.core. Moving the RaggedGather op kernel. PiperOrigin-RevId: 216400726 --- tensorflow/core/kernels/BUILD | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) (limited to 'tensorflow/core/kernels/BUILD') 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"], -- cgit v1.2.3