Searched refs:ipiv_array (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 1999 ipiv_array, in getrsBatched()
2015 ipiv_array, in getrsBatched()
2031 ipiv_array, in getrsBatched()
2047 ipiv_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()
2115 int* ipiv_array, in getrfBatched() argument
2124 ipiv_array, in getrfBatched()
[all …]
|
| H A D | CUDABlas.h | 290 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
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 101 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize,
284 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched() argument
286 …magma_dgetrf_batched(m, n, dA_array, ldda, ipiv_array, info_array, batchsize, magma_queue.get_queu… in magmaLuBatched()
293 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched() argument
295 …magma_sgetrf_batched(m, n, dA_array, ldda, ipiv_array, info_array, batchsize, magma_queue.get_queu… in magmaLuBatched()
302 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched() argument
304 …magma_zgetrf_batched(m, n, reinterpret_cast<magmaDoubleComplex**>(dA_array), ldda, ipiv_array, inf… in magmaLuBatched()
311 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched() argument
313 …magma_cgetrf_batched(m, n, reinterpret_cast<magmaFloatComplex**>(dA_array), ldda, ipiv_array, info… in magmaLuBatched()
|