Searched defs:ipiv_array (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 2096 int n, double** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched()
2104 int n, float** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched()
2115 int* ipiv_array, in getrfBatched()
2134 int* ipiv_array, in getrfBatched()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 284 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
293 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
302 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
311 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
|