Add StringLength to Flex delegate

StringLength op is used for tf.text models. PiperOrigin-RevId: 346531180 Change-Id: I2ce149fc5c191e799d11ad836c8f0205cf62610c
2020-12-09 05:38:11 -08:00 · 2020-12-09 05:38:11 -08:00 · 4ff4fbec62
commit 4ff4fbec62
parent 80df7cb64d
2 changed files with 2 additions and 0 deletions
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6262,6 +6262,7 @@ filegroup(
        "stateless_random_ops.cc",
        "stateless_random_ops_v2.cc",
        "string_join_op.cc",
+        "string_length_op.cc",
        "string_lower_op.cc",
        "string_util.cc",
        "string_split_op.cc",
--- a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@ -548,6 +548,7 @@ const std::set<std::string>& GetFlexAllowlist() {
          "StridedSliceAssign",
          "StridedSliceGrad",
          "StringJoin",
+          "StringLength",
          "StringLower",
          "StringSplit",
          "StringSplitV2",