Fix test to pass on GPU

tensorflower-gardener · tensorflower-gardener · commit 99b1aac98bb9 · 2024-09-10T16:21:28.000-07:00
PiperOrigin-RevId: 673138979
diff --git a/tensorflow/python/kernel_tests/signal/fft_ops_test.py b/tensorflow/python/kernel_tests/signal/fft_ops_test.py
@@ -627,7 +627,7 @@ def testFftLength_rfftn(self, dims, size, np_rtype):
     else:
       fft_length = (size * 2, size, size * 2)
       axes = (-3, -2, -1)
-    self._CompareBackward_fftn(
+    self._CompareForward_fftn(
         r2c.astype(np_rtype),
         fft_length=fft_length,
         axes=axes,
@@ -639,7 +639,7 @@ def testFftLength_rfftn(self, dims, size, np_rtype):
     c2r = self._generate_valid_irfft_input(
         c2r, np_ctype, r2c, np_rtype, 2, fft_length
     )
-    self._CompareForward_fftn(c2r, fft_length, axes, rtol=tol)
+    self._CompareBackward_fftn(c2r, fft_length, axes, rtol=tol)
 
   @parameterized.parameters(
       itertools.product(range(1, 4), (5, 6), (np.float32, np.float64))
