From 73edd2fb7e0ea40c9384824bc95c06478c789a01 Mon Sep 17 00:00:00 2001
From: Thai Nguyen <thaink@google.com>
Date: Wed, 3 Jun 2020 18:55:32 -0700
Subject: [PATCH] Support StringToHashBucketFast, Real, Imag ops in TFLite flex
 delegate

PiperOrigin-RevId: 314647136
Change-Id: I3e889044e961e22f26c63d00bd3a90083bb15330
---
 tensorflow/core/kernels/BUILD                          | 4 ++++
 tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
index 47b2fa44f57..fd6a8ab1cf6 100644
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@@ -6671,6 +6671,7 @@ filegroup(
         "spectrogram.h",
         "stateless_random_ops.h",
         "string_util.h",
+        "string_to_hash_bucket_op.h",
         "tensor_array.h",
         "tile_functor.h",
         "tile_ops_cpu_impl.h",
@@ -6735,6 +6736,7 @@ filegroup(
         "cwise_op_floor_mod.cc",
         "cwise_op_greater.cc",
         "cwise_op_greater_equal.cc",
+        "cwise_op_imag.cc",
         "cwise_op_invert.cc",
         "cwise_op_isfinite.cc",
         "cwise_op_isnan.cc",
@@ -6751,6 +6753,7 @@ filegroup(
         "cwise_op_mul_2.cc",
         "cwise_op_neg.cc",
         "cwise_op_pow.cc",
+        "cwise_op_real.cc",
         "cwise_op_reciprocal.cc",
         "cwise_op_right_shift.cc",
         "cwise_op_round.cc",
@@ -6902,6 +6905,7 @@ filegroup(
         "stateless_random_ops.cc",
         "string_join_op.cc",
         "string_util.cc",
+        "string_to_hash_bucket_op.cc",
         "summary_op.cc",
         "tensor_array.cc",
         "tensor_array_ops.cc",
diff --git a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
index d9150698298..e173b47ec32 100644
--- a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
@@ -163,6 +163,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
           "IRFFT",
           "IRFFT2D",
           "IRFFT3D",
+          "Imag",
           "ImmutableConst",
           "InTopK",
           "InTopKV2",
@@ -283,6 +284,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
           "RandomUniformInt",
           "Range",
           "Rank",
+          "Real",
           "RealDiv",
           "Reciprocal",
           "ReciprocalGrad",
@@ -423,6 +425,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
           "StridedSliceAssign",
           "StridedSliceGrad",
           "StringJoin",
+          "StringToHashBucketFast",
           "Sub",
           "Sum",
           "Switch",