Searched refs:a_comp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 112 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()() local
119 csr_geam.GetWorkspaceSize(a_comp, b_comp, &thisWorkspaceSize)); in operator ()()
137 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()() local
145 a_comp, b_comp, c_row_ptr_i, &c_nnz_i, workspace)); in operator ()()
167 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()() local
174 TF_RETURN_IF_ERROR(csr_geam.Compute(a_comp, b_comp, &c_comp, workspace)); in operator ()()
|
| D | sparse_mat_mul_op.cc | 424 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute() local
434 ctx, csr_gemm.GetWorkspaceSize(a_comp, b_comp, &thisWorkspaceSize)); in Compute()
453 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute() local
467 csr_gemm.GetOutputStructure(a_comp, b_comp, c_row_ptr_i, in Compute()
488 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute() local
498 OP_REQUIRES_OK(ctx, csr_gemm.Compute(a_comp, b_comp, &c_comp, workspace)); in Compute()
|
| D | mat_mul_op.cc | 577 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute() local
581 Status s = csr_spmv.Compute(ctx, a_comp, b_i, c_i); in Compute()
672 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute() local
674 Status s = csr_spmmadd.Compute(ctx, a_comp, b_i, c_mat_col_major_i); in Compute()
|