Searched refs:CSRSparseSparseMatrixMatMul (Results 1 – 2 of 2) sorted by relevance
194 struct CSRSparseSparseMatrixMatMul struct196 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a,
375 functor::CSRSparseSparseMatrixMatMul<Device, T> csr_gemm( in Compute()552 struct CSRSparseSparseMatrixMatMul<GPUDevice, T> struct554 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a, in CSRSparseSparseMatrixMatMul() argument577 ~CSRSparseSparseMatrixMatMul() { in CSRSparseSparseMatrixMatMul() argument