Home
last modified time | relevance | path

Searched defs:signal_numel (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DCuFFTPlanCache.h168 …const auto signal_numel = c10::multiply_integers(sizes.slice(1, sizes.size() - 2)) * last_dim_size; in as_cufft_embed() local
DSpectralOps.cpp297 int64_t signal_numel = 1; in _fft_normalization_scale() local
/external/pytorch/aten/src/ATen/native/mkl/
DSpectralOps.cpp408 int64_t signal_numel = c10::multiply_integers(IntArrayRef(sizes.data() + 1, signal_ndim)); in _plan_mkl_fft() local