Add set kernels to Flex

closes tensorflow#45199 PiperOrigin-RevId: 351490161 Change-Id: I6ce7e52a22c16d34ae007e448f5ebd6482e3556a
CTouch · Jan 13, 2021 · 103ff00 · 103ff00
1 parent 3654281
commit 103ff00
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
@@ -6195,6 +6195,7 @@ filegroup(
         "segment_reduction_ops_impl_4.cc",
         "segment_reduction_ops_impl_5.cc",
         "session_ops.cc",
+        "set_kernels.cc",
         "softplus_op.cc",
         "softsign_op.cc",
         "spacetobatch_functor.cc",

diff --git a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@@ -146,6 +146,8 @@ const std::set<std::string>& GetFlexAllowlist() {
           "DeepCopy",
           "DeleteSessionTensor",
           "DenseBincount",
+          "DenseToDenseSetOperation",
+          "DenseToSparseSetOperation",
           "DepthToSpace",
           "DepthwiseConv2dNative",
           "Dequantize",
@@ -512,6 +514,7 @@ const std::set<std::string>& GetFlexAllowlist() {
           "SparseSegmentSum",
           "SparseSegmentSumWithNumSegments",
           "SparseToDense",
+          "SparseToSparseSetOperation",
           "Split",
           "SplitV",
           "Sqrt",