Do not normalize FFT by a constant "1" if no normalization is provided (i.e., norm is None).
Do not normalize FFT by a constant "1" if no normalization is provided (i.e., norm is None).

Without this, the compiled graph will still contain a node multipying a complex number with a constant 1+0j (1 is cast to complex because the other term is complex as well). This is problematic when converting to TFLite using jax2tf, because multiplying complex numbers is not supported in TFLite. With this change, the multiplication is removed from the graph all together.

PiperOrigin-RevId: 459566727

google/jax,google/jax,google/jax,google/jax