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`