Searched refs:A_ptr_array_data (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| D | BatchLinearAlgebraLibBlas.cpp | 213 auto A_ptr_array_data = reinterpret_cast<scalar_t**>(A_ptr_array.data_ptr()); in apply_triangular_solve_batched() local
217 …at::cuda::blas::trsmBatched(handle, side, uplo, trans, diag, m, n, &alpha, A_ptr_array_data, lda, … in apply_triangular_solve_batched()
275 auto A_ptr_array_data = reinterpret_cast<scalar_t**>(A_ptr_array.data_ptr()); in apply_gels_batched() local
284 A_ptr_array_data, lda, in apply_gels_batched()
|