Home
last modified time | relevance | path

Searched refs:dA_array (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp100 magma_int_t m, magma_int_t n, scalar_t** dA_array, magma_int_t ldda,
111 magma_int_t m, magma_int_t n, scalar_t** dA_array, magma_int_t ldda,
121 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, scalar_t** dA_array, magma_int_t ldda,
131 magma_uplo_t uplo, magma_int_t n, scalar_t** dA_array, magma_int_t ldda,
137 …scalar_t** dA_array, magma_int_t ldda, scalar_t** dB_array, magma_int_t lddb, magma_int_t batchsiz…
177 … magma_int_t n, magma_int_t nrhs, scalar_t** dA_array, magma_int_t ldda, magma_int_t** dipiv_array,
283 magma_int_t m, magma_int_t n, double** dA_array, magma_int_t ldda, in magmaLuBatched() argument
286 …magma_dgetrf_batched(m, n, dA_array, ldda, ipiv_array, info_array, batchsize, magma_queue.get_queu… in magmaLuBatched()
292 magma_int_t m, magma_int_t n, float** dA_array, magma_int_t ldda, in magmaLuBatched() argument
295 …magma_sgetrf_batched(m, n, dA_array, ldda, ipiv_array, info_array, batchsize, magma_queue.get_queu… in magmaLuBatched()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp1997 dA_array, in getrsBatched()
2013 dA_array, in getrsBatched()
2029 reinterpret_cast<cuComplex**>(dA_array), in getrsBatched()
2045 reinterpret_cast<cuDoubleComplex**>(dA_array), in getrsBatched()
2096 int n, double** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched() argument
2099 handle, n, dA_array, ldda, ipiv_array, info_array, batchsize)); in getrfBatched()
2104 int n, float** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched() argument
2107 handle, n, dA_array, ldda, ipiv_array, info_array, batchsize)); in getrfBatched()
2113 c10::complex<double>** dA_array, in getrfBatched() argument
2122 reinterpret_cast<cuDoubleComplex**>(dA_array), in getrfBatched()
[all …]
H A DCUDABlas.h290 int n, int nrhs, Dtype** dA_array, int lda, int* ipiv_array, \
326 int n, Dtype** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize
342 …cublasHandle_t handle, cublasOperation_t trans, int m, int n, int nrhs, Dtype** dA_array, int ldda…