Searched defs:dB_array (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 428 …double** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const MAGMAQueue& m… in magmaCholeskySolveBatched()
436 …float** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const MAGMAQueue& ma… in magmaCholeskySolveBatched()
444 …c10::complex<double>** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const… in magmaCholeskySolveBatched()
454 …c10::complex<float>** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const … in magmaCholeskySolveBatched()
532 double** dA_array, magma_int_t ldda, double** dB_array, magma_int_t lddb, magma_int_t batchsize, in magmaTriangularSolveBatched()
541 float** dA_array, magma_int_t ldda, float** dB_array, magma_int_t lddb, magma_int_t batchsize, in magmaTriangularSolveBatched()
550 …c10::complex<double>** dA_array, magma_int_t ldda, c10::complex<double>** dB_array, magma_int_t ld… in magmaTriangularSolveBatched()
562 …c10::complex<float>** dA_array, magma_int_t ldda, c10::complex<float>** dB_array, magma_int_t lddb… in magmaTriangularSolveBatched()
906 double** dB_array, magma_int_t lddb, magma_int_t& info, in magmaLuSolveBatched()
915 float** dB_array, magma_int_t lddb, magma_int_t& info, in magmaLuSolveBatched()
[all …]
|