Reciprocal is not executed on GPU for complex number

[WindQAQ]

Please make sure that this is a bug. As per our
GitHub Policy,
we only address code/doc bugs, performance issues, feature requests and
build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow):
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device:
• TensorFlow installed from (source or binary):
• TensorFlow version (use command below):
• Python version:
• Bazel version (if compiling from source):
• GCC/Compiler version (if compiling from source):
• CUDA/cuDNN version: colab GPU
• GPU model and memory: colab GPU

You can collect some of this information using our environment capture
script
You can also obtain the TensorFlow version with:

1. TF 1.0: python -c "import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"
2. TF 2.0: python -c "import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"

Describe the current behavior

Reciprocal is not executed on GPU for complex number.

import tensorflow as tf

tf.debugging.set_log_device_placement(True)
with tf.device("gpu"):
  tf.math.reciprocal(tf.complex(1.0, 0.0))

"""
Executing op Complex in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op Reciprocal in device /job:localhost/replica:0/task:0/device:CPU:0
"""

Describe the expected behavior

The reciprocal op can be executed on GPU for complex number

Standalone code to reproduce the issue
Provide a reproducible test case that is the bare minimum necessary to generate
the problem. If possible, please share a link to Colab/Jupyter/any notebook.

https://colab.research.google.com/drive/1x0NX11ZBjA95KSeafLg0_LRvVe8L2uyD?usp=sharing

Other info / logs Include any logs or source code that would be helpful to
diagnose the problem. If including tracebacks, please include the full
traceback. Large logs and files should be attached.

[ravikyram]

I have tried in colab with TF version 2.2, 2.3-rc1,nightly versions(2.4.0-dev20200719) and was able to reproduce the issue.Please, find the gist here.Thanks!

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No