Add Conj op to flex's whitelist

PiperOrigin-RevId: 306943019 Change-Id: I24c1dd2ac20824506dcbe241ac71cdc9e552b78e
2020-04-16 16:22:06 -07:00 · 2020-04-16 16:22:06 -07:00 · 1bbc26d375
commit 1bbc26d375
parent d5743e269c
2 changed files with 2 additions and 0 deletions
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6654,6 +6654,7 @@ filegroup(
        "cwise_op_bitwise_and.cc",
        "cwise_op_bitwise_or.cc",
        "cwise_op_bitwise_xor.cc",
+        "cwise_op_conj.cc",
        "cwise_op_cos.cc",
        "cwise_op_cosh.cc",
        "cwise_op_div.cc",
--- a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
@ -76,6 +76,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
          "Concat",
          "ConcatOffset",
          "ConcatV2",
+          "Conj",
          "ConjugateTranspose",
          "Const",
          "ControlTrigger",