Method _to_expected

tensorflow/compiler/tests/fft_test.py:135–136

(x)

  def testRFFT(self):

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

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