Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DCUDASolver.h268 void geqrf_bufferSize(CUDASOLVER_GEQRF_BUFFERSIZE_ARGTYPES(scalar_t)) { in geqrf_bufferSize() function
H A DCUDASolver.cpp715 void geqrf_bufferSize<float>(CUDASOLVER_GEQRF_BUFFERSIZE_ARGTYPES(float)) { in geqrf_bufferSize() function
721 void geqrf_bufferSize<double>(CUDASOLVER_GEQRF_BUFFERSIZE_ARGTYPES(double)) { in geqrf_bufferSize() function
727 void geqrf_bufferSize<c10::complex<float>>( in geqrf_bufferSize() function
734 void geqrf_bufferSize<c10::complex<double>>( in geqrf_bufferSize() function