Searched refs:a_input_matrix (Results 1 – 2 of 2) sorted by relevance
391 const CSRSparseMatrix* a_input_matrix; in Compute() local396 a_input_matrix = a_matrix; in Compute()401 a_input_matrix = &a_matrix_transposed; in Compute()403 auto a_input_dense_shape = a_input_matrix->dense_shape().vec<int64>(); in Compute()423 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()424 a_input_matrix->col_indices_vec(i), in Compute()425 a_input_matrix->values_vec<T>(i), in Compute()457 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()458 a_input_matrix->col_indices_vec(i), in Compute()459 a_input_matrix->values_vec<T>(i), in Compute()
614 const CSRSparseMatrix* a_input_matrix; in Compute() local619 a_input_matrix = a_matrix; in Compute()624 a_input_matrix = &a_matrix_transposed; in Compute()627 auto a_input_dense_shape = a_input_matrix->dense_shape().vec<int64>(); in Compute()659 auto a_row_ptr = a_input_matrix->row_pointers_vec(i); in Compute()660 auto a_col_ind = a_input_matrix->col_indices_vec(i); in Compute()661 auto a_values = a_input_matrix->values_vec<T>(i); in Compute()