Add RaggedGather op to Flex delegate

PiperOrigin-RevId: 335776475 Change-Id: I3c96402141609f29f363f84afbe029edd8da0ec7
2020-10-06 19:58:49 -07:00 · 2020-10-06 19:58:49 -07:00 · e6021dc422
commit e6021dc422
parent 397d9c8ef9
2 changed files with 2 additions and 0 deletions
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6196,6 +6196,7 @@ filegroup(
        "queue_op.cc",
        "queue_ops.cc",
        "ragged_range_op.cc",
+        "ragged_gather_op.cc",
        "ragged_tensor_to_sparse_kernel.cc",
        "ragged_tensor_to_tensor_op.cc",
        "random_op.cc",
--- a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@ -354,6 +354,7 @@ const std::set<std::string>& GetFlexAllowlist() {
          "RFFT2D",
          "RFFT3D",
          "RaggedBincount",
+          "RaggedGather",
          "RaggedRange",
          "RaggedTensorToSparse",
          "RaggedTensorToTensor",