Home
last modified time | relevance | path

Searched defs:signal_ndim (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/mkl/
DDescriptors.h19 …void init(DFTI_CONFIG_VALUE precision, DFTI_CONFIG_VALUE signal_type, MKL_LONG signal_ndim, MKL_LO… in init()
/external/pytorch/aten/src/ATen/native/cuda/
DCuFFTPlanCache.h161 const auto signal_ndim = strides.size() - 1; in as_cufft_embed() local
251 const int64_t signal_ndim = sizes.size() - 1; in CuFFTConfig() local
DSpectralOps.cpp169 const int64_t signal_ndim = dim.size(); in _exec_fft() local
/external/pytorch/aten/src/ATen/native/mkl/
DSpectralOps.cpp353 const int64_t signal_ndim = sizes.size() - 1; in _plan_mkl_fft() local
433 const int64_t signal_ndim = dim.size(); in _exec_fft() local
/external/pytorch/test/
Dtest_spectral_ops.py767 def _test_complex(sizes, signal_ndim, prepro_fn=lambda x: x): argument
778 def _test_real(sizes, signal_ndim, prepro_fn=lambda x: x): argument