Searched refs:a_comp (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | add_op.cc | 117 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 ()()
|
D | sparse_mat_mul_op.cc | 423 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()
|
D | mat_mul_op.cc | 572 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()
|