Searched refs:shape_from_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SpectralOps.cpp | 221 inline shape_t shape_from_tensor(const Tensor& t) { in shape_from_tensor() function
268 …pocketfft::c2r(shape_from_tensor(out), stride_from_tensor(self), stride_from_tensor(out), axes, fa… in _fft_c2r_mkl()
272 …pocketfft::c2r(shape_from_tensor(out), stride_from_tensor(self), stride_from_tensor(out), axes, fa… in _fft_c2r_mkl()
293 …pocketfft::r2c(shape_from_tensor(self), stride_from_tensor(self), stride_from_tensor(out), axes, t… in _fft_r2c_mkl()
297 …pocketfft::r2c(shape_from_tensor(self), stride_from_tensor(self), stride_from_tensor(out), axes, t… in _fft_r2c_mkl()
317 …pocketfft::c2c(shape_from_tensor(self), stride_from_tensor(self), stride_from_tensor(out), axes, f… in _fft_c2c_mkl()
321 …pocketfft::c2c(shape_from_tensor(self), stride_from_tensor(self), stride_from_tensor(out), axes, f… in _fft_c2c_mkl()
|