Add tf.StringToNumber op to Flex

PiperOrigin-RevId: 356642595 Change-Id: I57a7ffac9fdfaba5ac92a65e468d2d1bf7b4d6a7
2021-02-09 18:57:00 -08:00 · 2021-02-09 18:57:00 -08:00 · 02e6913295
commit 02e6913295
parent d406c45eb1
3 changed files with 3 additions and 0 deletions
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6233,6 +6233,7 @@ filegroup(
        "string_strip_op.cc",
        "string_to_hash_bucket_op.cc",
        "string_to_hash_bucket_fast_op.cc",
+        "string_to_number_op.cc",
        "substr_op.cc",
        "tensor_array.cc",
        "tensor_array_ops.cc",
--- a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@ -588,6 +588,7 @@ const std::set<std::string>& GetFlexAllowlist() {
          "StringToHashBucket",
          "StringToHashBucketFast",
          "StringToHashBucketStrong",
+          "StringToNumber",
          "Sub",
          "Substr",
          "Sum",
--- a/tensorflow/lite/g3doc/guide/op_select_allowlist.md
+++ b/tensorflow/lite/g3doc/guide/op_select_allowlist.md
@ -565,6 +565,7 @@ supported by TensorFlow Lite runtime with the Select TensorFlow Ops feature.
 *   `raw_ops.StringToHashBucket`
 *   `raw_ops.StringToHashBucketFast`
 *   `raw_ops.StringToHashBucketStrong`
+*   `raw_ops.StringToNumber`
 *   `raw_ops.Sub`
 *   `raw_ops.Substr`
 *   `raw_ops.Sum`