Searched defs:CSRSparseSparseMatrixMatMul (Results 1 – 2 of 2) sorted by relevance
552 struct CSRSparseSparseMatrixMatMul<GPUDevice, T> struct554 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a, in CSRSparseSparseMatrixMatMul() argument577 ~CSRSparseSparseMatrixMatMul() { in CSRSparseSparseMatrixMatMul()584 Status Initialize() { in CSRSparseSparseMatrixMatMul()601 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul()630 Status GetOutputStructure(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul()675 Status Compute(const ConstCSRComponent<T>& a, const ConstCSRComponent<T>& b, in CSRSparseSparseMatrixMatMul()726 OpKernelContext* ctx_; in CSRSparseSparseMatrixMatMul()727 GpuSparse cuda_sparse_; in CSRSparseSparseMatrixMatMul()728 bool initialized_; in CSRSparseSparseMatrixMatMul()[all …]
194 struct CSRSparseSparseMatrixMatMul struct