Add support for complex number scale values

[tlemo]

This is a fix required for https://github.com/tensorflow/tensorflow/pull/52949: for complex numbers, scale type must match element type, not compute type.