Method _tf_fn

tensorflow/compiler/tests/fft_test.py:138–139

(x)

      return np.fft.rfft(x, n=x.shape[-1])

    def _tf_fn(x):
      return signal.rfft(x, fft_length=[x.shape[-1]])

    self._VerifyFftMethod(INNER_DIMS_1D, np.real, _to_expected, _tf_fn)