Home
last modified time | relevance | path

Searched refs:a_input_matrix (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc391 const CSRSparseMatrix* a_input_matrix; in Compute() local
396 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()
Dmat_mul_op.cc614 const CSRSparseMatrix* a_input_matrix; in Compute() local
619 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()