Searched refs:tensor_cdata (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SpectralOps.cpp | 226 inline std::complex<T> *tensor_cdata(Tensor& t) { in tensor_cdata() function
231 inline const std::complex<T> *tensor_cdata(const Tensor& t) { in tensor_cdata() function
269 tensor_cdata<float>(self), in _fft_c2r_mkl()
273 tensor_cdata<double>(self), in _fft_c2r_mkl()
295 tensor_cdata<float>(out), compute_fct<float>(self, dim, normalization)); in _fft_r2c_mkl()
299 tensor_cdata<double>(out), compute_fct<double>(self, dim, normalization)); in _fft_r2c_mkl()
318 tensor_cdata<float>(self), in _fft_c2c_mkl()
319 tensor_cdata<float>(out), compute_fct<float>(self, dim, normalization)); in _fft_c2c_mkl()
322 tensor_cdata<double>(self), in _fft_c2c_mkl()
323 tensor_cdata<double>(out), compute_fct<double>(self, dim, normalization)); in _fft_c2c_mkl()
|