Searched refs:prepare_column_major_matrix_for_cusparse (Results 1 – 1 of 1) sorted by relevance
30 c10::MaybeOwned<Tensor> prepare_column_major_matrix_for_cusparse( in prepare_column_major_matrix_for_cusparse() function43 return prepare_column_major_matrix_for_cusparse(tensor); in prepare_dense_matrix_for_cusparse()293 c10::MaybeOwned<Tensor> X_ = prepare_column_major_matrix_for_cusparse(X); in block_sparse_triangular_solve_mat()294 c10::MaybeOwned<Tensor> B_ = prepare_column_major_matrix_for_cusparse(B); in block_sparse_triangular_solve_mat()512 prepare_column_major_matrix_for_cusparse(result); in block_sparse_mm()