diff options
author | Haoliang Zhang <haoliang@google.com> | 2021-02-17 15:50:50 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2021-02-17 15:59:18 -0800 |
commit | 03000e093c49199b524e331076ad81912ca3d836 (patch) | |
tree | 5e048578d67146f9828f8a07ed215e747db9fcf9 | |
parent | 8252d1c7888e4a0dcae7129d1c8cb3e3635fb5ea (diff) | |
download | tensorflow-03000e093c49199b524e331076ad81912ca3d836.tar.gz |
Add `StringFormat` to allowed flex ops.
PiperOrigin-RevId: 358051221
Change-Id: Ieea2db1bc27e80d7197b0c60fd7177ba41992aba
-rw-r--r-- | tensorflow/core/kernels/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc | 1 | ||||
-rw-r--r-- | tensorflow/lite/g3doc/guide/op_select_allowlist.md | 1 |
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` |