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

Method testRFFT3DMismatchedSize

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

Source from the content-addressed store, hash-verified

162 self._VerifyFftMethod(INNER_DIMS_3D, np.real, _to_expected, _tf_fn)
163
164 def testRFFT3DMismatchedSize(self):
165
166 def _to_expected(x):
167 return np.fft.rfftn(
168 x,
169 axes=(-3, -2, -1),
170 s=[x.shape[-3] // 2, x.shape[-2], x.shape[-1] * 2])
171
172 def _tf_fn(x):
173 return signal.rfft3d(
174 x, fft_length=[x.shape[-3] // 2, x.shape[-2], x.shape[-1] * 2])
175
176 self._VerifyFftMethod(INNER_DIMS_3D, np.real, _to_expected, _tf_fn)
177
178 def testIRFFT(self):
179

Callers

nothing calls this directly

Calls 1

_VerifyFftMethodMethod · 0.95

Tested by

no test coverage detected