Home
last modified time | relevance | path

Searched refs:a_comp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc117 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()() local
125 csr_geam.GetOutputStructure(a_comp, b_comp, c_row_ptr_i, &c_nnz_i)); in operator ()()
147 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()() local
154 TF_RETURN_IF_ERROR(csr_geam.Compute(a_comp, b_comp, &c_comp)); in operator ()()
Dsparse_mat_mul_op.cc423 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute() local
436 OP_REQUIRES_OK(ctx, csr_gemm.GetOutputStructure(a_comp, b_comp, in Compute()
457 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute() local
467 OP_REQUIRES_OK(ctx, csr_gemm.Compute(a_comp, b_comp, &c_comp)); in Compute()
Dmat_mul_op.cc572 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute() local
576 Status s = csr_spmv.Compute(ctx, a_comp, b_i, c_i); in Compute()
667 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute() local
669 Status s = csr_spmmadd.Compute(ctx, a_comp, b_i, c_mat_col_major_i); in Compute()