Lines Matching defs:device
116 def skip_helper_for_fft(device, dtype): argument
134 def test_reference_1d(self, device, dtype, op): argument
179 def test_fft_round_trip(self, device, dtype): argument
236 def test_empty_fft(self, device, dtype, op): argument
244 def test_empty_ifft(self, device): argument
254 def test_fft_invalid_dtypes(self, device): argument
270 def test_fft_type_promotion(self, device, dtype): argument
322 def test_fft_half_and_bfloat16_errors(self, device, dtype, op): argument
338 def test_fft_half_and_chalf_not_power_of_two_error(self, device, dtype, op): argument
357 def test_reference_nd(self, device, dtype, op): argument
393 def test_fftn_round_trip(self, device, dtype): argument
444 def test_fftn_invalid(self, device, dtype, op): argument
466 def test_fftn_noop_transform(self, device, dtype): argument
494 def test_hfftn(self, device, dtype): argument
533 def test_ihfftn(self, device, dtype): argument
575 def test_fft2_numpy(self, device, dtype): argument
621 def test_fft2_fftn_equivalence(self, device, dtype): argument
657 def test_fft2_invalid(self, device): argument
685 def test_fftfreq_numpy(self, device, dtype): argument
710 def test_fftfreq_out(self, device, dtype): argument
723 def test_fftshift_numpy(self, device, dtype): argument
749 def test_fftshift_frequencies(self, device, dtype): argument
764 def _test_fft_ifft_rfft_irfft(self, device, dtype): argument
823 def test_fft_ifft_rfft_irfft(self, device, dtype): argument
831 def plan_cache_max_size(device, n): argument
902 def test_cufft_context(self, device, dtype): argument
921 def test_stft(self, device, dtype): argument
995 def test_istft_against_librosa(self, device, dtype): argument
1041 def test_complex_stft_roundtrip(self, device, dtype): argument
1083 def test_stft_roundtrip_complex_window(self, device, dtype): argument
1123 def test_complex_stft_definition(self, device, dtype): argument
1143 def test_complex_stft_real_equiv(self, device, dtype): argument
1176 def test_complex_istft_real_equiv(self, device, dtype): argument
1201 def test_complex_stft_onesided(self, device): argument
1223 def test_stft_requires_complex(self, device): argument
1231 def test_stft_requires_window(self, device): argument
1238 def test_istft_requires_window(self, device): argument
1245 def test_fft_input_modification(self, device): argument
1266 def test_fft_plan_repeatable(self, device): argument
1282 def test_istft_round_trip_simple_cases(self, device, dtype): argument
1295 def test_istft_round_trip_various_params(self, device, dtype): argument
1366 def test_istft_round_trip_with_padding(self, device, dtype): argument
1431 def test_istft_throws(self, device): argument
1450 def test_istft_of_sine(self, device, dtype): argument
1486 def test_istft_linearity(self, device, dtype): argument
1552 def test_batch_istft(self, device): argument
1571 def test_stft_window_device(self, device): argument
1623 def test(self, device): argument