Searched defs:dA_array (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 283 magma_int_t m, magma_int_t n, double** dA_array, magma_int_t ldda, in magmaLuBatched()
292 magma_int_t m, magma_int_t n, float** dA_array, magma_int_t ldda, in magmaLuBatched()
301 magma_int_t m, magma_int_t n, c10::complex<double>** dA_array, magma_int_t ldda, in magmaLuBatched()
310 magma_int_t m, magma_int_t n, c10::complex<float>** dA_array, magma_int_t ldda, in magmaLuBatched()
355 magma_int_t m, magma_int_t n, double** dA_array, magma_int_t ldda, in magmaLuNoPivBatched()
363 magma_int_t m, magma_int_t n, float** dA_array, magma_int_t ldda, in magmaLuNoPivBatched()
371 magma_int_t m, magma_int_t n, c10::complex<double>** dA_array, magma_int_t ldda, in magmaLuNoPivBatched()
379 magma_int_t m, magma_int_t n, c10::complex<float>** dA_array, magma_int_t ldda, in magmaLuNoPivBatched()
427 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, double** dA_array, magma_int_t ldda, in magmaCholeskySolveBatched()
435 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, float** dA_array, magma_int_t ldda, in magmaCholeskySolveBatched()
[all …]
|
| /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()
2113 c10::complex<double>** dA_array, in getrfBatched()
2132 c10::complex<float>** dA_array, in getrfBatched()
|