MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / testIRFFT2D

Method testIRFFT2D

tensorflow/compiler/tests/fft_test.py:187–196  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

185 lambda x: np.fft.irfft(x, n=2 * (x.shape[-1] - 1)), _tf_fn)
186
187 def testIRFFT2D(self):
188
189 def _tf_fn(x):
190 return signal.irfft2d(x, fft_length=[x.shape[-2], 2 * (x.shape[-1] - 1)])
191
192 self._VerifyFftMethod(
193 INNER_DIMS_2D,
194 lambda x: np.fft.rfft2(np.real(x), s=[x.shape[-2], x.shape[-1]]),
195 lambda x: np.fft.irfft2(x, s=[x.shape[-2], 2 * (x.shape[-1] - 1)]),
196 _tf_fn)
197
198 def testIRFFT3D(self):
199

Callers

nothing calls this directly

Calls 1

_VerifyFftMethodMethod · 0.95

Tested by

no test coverage detected