Error when executing tf.math.conj or tf.norm for complex data on GPU

Question

When executing tf.math.conj or tf.norm on complex data in a Quadro RTX 5000 GPU, it results in a CUDA_ERROR_LAUNCH_FAILED error.

This is the code re