aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHaoliang Zhang <haoliang@google.com>2021-02-17 15:50:50 -0800
committerTensorFlower Gardener <gardener@tensorflow.org>2021-02-17 15:59:18 -0800
commit03000e093c49199b524e331076ad81912ca3d836 (patch)
tree5e048578d67146f9828f8a07ed215e747db9fcf9
parent8252d1c7888e4a0dcae7129d1c8cb3e3635fb5ea (diff)
downloadtensorflow-03000e093c49199b524e331076ad81912ca3d836.tar.gz
Add `StringFormat` to allowed flex ops.
PiperOrigin-RevId: 358051221 Change-Id: Ieea2db1bc27e80d7197b0c60fd7177ba41992aba
-rw-r--r--tensorflow/core/kernels/BUILD1
-rw-r--r--tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc1
-rw-r--r--tensorflow/lite/g3doc/guide/op_select_allowlist.md1
3 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
index 1b353622261..198b3ee5baf 100644
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@@ -6227,6 +6227,7 @@ filegroup(
"stateless_random_gamma_op.cc",
"stateless_random_ops.cc",
"stateless_random_ops_v2.cc",
+ "string_format_op.cc",
"string_join_op.cc",
"string_length_op.cc",
"string_lower_op.cc",
diff --git a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
index 95af8894848..e53e9f7b5cd 100644
--- a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@@ -579,6 +579,7 @@ const std::set<std::string>& GetFlexAllowlist() {
"StridedSlice",
"StridedSliceAssign",
"StridedSliceGrad",
+ "StringFormat",
"StringJoin",
"StringLength",
"StringLower",
diff --git a/tensorflow/lite/g3doc/guide/op_select_allowlist.md b/tensorflow/lite/g3doc/guide/op_select_allowlist.md
index b9f3ac8b7df..6051b5b988d 100644
--- a/tensorflow/lite/g3doc/guide/op_select_allowlist.md
+++ b/tensorflow/lite/g3doc/guide/op_select_allowlist.md
@@ -556,6 +556,7 @@ supported by TensorFlow Lite runtime with the Select TensorFlow Ops feature.
* `raw_ops.StridedSlice`
* `raw_ops.StridedSliceAssign`
* `raw_ops.StridedSliceGrad`
+* `raw_ops.StringFormat`
* `raw_ops.StringJoin`
* `raw_ops.StringLength`
* `raw_ops.StringLower`