Home
last modified time | relevance | path

Searched refs:ConstCSRComponent (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels.h148 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a,
158 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a,
170 virtual Status GetWorkspaceSize(const ConstCSRComponent<T>& a,
171 const ConstCSRComponent<T>& b,
174 virtual Status GetOutputStructure(const ConstCSRComponent<T>& a,
175 const ConstCSRComponent<T>& b,
179 virtual Status Compute(const ConstCSRComponent<T>& a,
180 const ConstCSRComponent<T>& b, CSRComponent<T>* c,
203 Status operator()(OpKernelContext* ctx, const ConstCSRComponent<T>& x,
Dadd_op.cc112 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
114 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
137 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
139 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
167 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
169 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
292 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in GetWorkspaceSize()
293 const ConstCSRComponent<T>& b, size_t* bufferSize) { in GetWorkspaceSize()
318 Status GetOutputStructure(const ConstCSRComponent<T>& a, in GetOutputStructure()
319 const ConstCSRComponent<T>& b, in GetOutputStructure()
[all …]
Dsparse_mat_mul_op.cc424 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
428 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_vec(i), in Compute()
453 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
457 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_vec(i), in Compute()
488 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
492 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_vec(i), in Compute()
601 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul()
602 const ConstCSRComponent<T>& b, size_t* bufferSize) { in CSRSparseSparseMatrixMatMul()
630 Status GetOutputStructure(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul()
631 const ConstCSRComponent<T>& b, in CSRSparseSparseMatrixMatMul()
[all …]
Dtranspose_op.cc53 Status ValidateTransposeInputs(const ConstCSRComponent<T>& input, in ValidateTransposeInputs()
197 ConstCSRComponent<T> input_comp{ in operator ()()
219 Status operator()(OpKernelContext* ctx, const ConstCSRComponent<T>& input, in operator ()()
260 Status operator()(OpKernelContext* ctx, const ConstCSRComponent<T>& x, in operator ()()
Dmat_mul_op.cc577 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute()
672 ConstCSRComponent<T> a_comp{a_row_ptr, a_col_ind, a_values, in Compute()
778 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a, in Compute()
918 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a, in Compute()
Dsparse_matrix.h617 struct ConstCSRComponent { struct