Searched refs:A_ptr_array (Results 1 – 2 of 2) sorted by relevance
211 Tensor A_ptr_array = get_device_pointers<scalar_t>(A); in apply_triangular_solve_batched() local213 auto A_ptr_array_data = reinterpret_cast<scalar_t**>(A_ptr_array.data_ptr()); in apply_triangular_solve_batched()273 Tensor A_ptr_array = get_device_pointers<scalar_t>(A_broadcasted); in apply_gels_batched() local275 auto A_ptr_array_data = reinterpret_cast<scalar_t**>(A_ptr_array.data_ptr()); in apply_gels_batched()
892 auto A_ptr_array = get_device_pointers<scalar_t>(A_column_major_copy); in apply_cholesky_cusolver_potrsBatched() local898 reinterpret_cast<scalar_t**>(A_ptr_array.data_ptr()), in apply_cholesky_cusolver_potrsBatched()