Home
last modified time | relevance | path

Searched defs:last_dim_halfsize (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSpectralOps.cpp181 auto last_dim_halfsize = result.sizes()[last_dim]; in _fft_r2c_mkl_out() local
285 auto last_dim_halfsize = (input_sizes[last_dim]) / 2 + 1; in _fft_r2c_mkl() local
545 auto last_dim_halfsize = (input_sizes[last_dim]) / 2 + 1; in _fft_r2c_mkl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSpectralOps.cpp336 auto last_dim_halfsize = (input_sizes[last_dim]) / 2 + 1; in _fft_r2c_cufft() local
410 auto last_dim_halfsize = result.sizes()[last_dim]; in _fft_r2c_cufft_out() local