From 02e691329517eb5e76522ed8d8bef79ceb082ff8 Mon Sep 17 00:00:00 2001 From: Jaesung Chung Date: Tue, 9 Feb 2021 18:57:00 -0800 Subject: [PATCH] Add tf.StringToNumber op to Flex PiperOrigin-RevId: 356642595 Change-Id: I57a7ffac9fdfaba5ac92a65e468d2d1bf7b4d6a7 --- tensorflow/core/kernels/BUILD | 1 + tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc | 1 + tensorflow/lite/g3doc/guide/op_select_allowlist.md | 1 + 3 files changed, 3 insertions(+) diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 8a02136beea..a3037c22db8 100644 --- 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", diff --git a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc index 9c196d57664..95af8894848 100644 --- 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& GetFlexAllowlist() { "StringToHashBucket", "StringToHashBucketFast", "StringToHashBucketStrong", + "StringToNumber", "Sub", "Substr", "Sum", diff --git a/tensorflow/lite/g3doc/guide/op_select_allowlist.md b/tensorflow/lite/g3doc/guide/op_select_allowlist.md index 0ab5adba9b0..b9f3ac8b7df 100644 --- 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`