Commit 8f0a90b7 authored by RJ Ryan's avatar RJ Ryan Committed by TensorFlower Gardener
Browse files

Add complex128 support to FFT, FFT2D, FFT3D, IFFT, IFFT2D, and IFFT3D. 

NumPy automatically upcasts to complex128 when computing FFTs, leading to issues like:
#10749

This change allows users to choose between 32-bit and 64-bit precision FFTs on CPU and GPU.

PiperOrigin-RevId: 195183206
parent 7833890a
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment