Home
last modified time | relevance | path

Searched refs:GetCuFFTTransformType (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DCuFFTPlanCache.h90 inline CuFFTTransformType GetCuFFTTransformType(bool complex_input, bool complex_output) { in GetCuFFTTransformType() function
DSpectralOps.cpp217 auto fft_type = GetCuFFTTransformType(input.is_complex(), out.is_complex()); in _exec_fft()