Home
last modified time | relevance | path

Searched defs:CusparseMatrixMultiplyOp (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseMatMul.cu204 struct CusparseMatrixMultiplyOp { struct
206 cusparseSpGEMMDescr_t spgemmDesc;
208 CusparseMatrixMultiplyOp() { in CusparseMatrixMultiplyOp() argument
221 ~CusparseMatrixMultiplyOp() { in ~CusparseMatrixMultiplyOp()
226 csrOutput operator ()( in operator ()()
383 template struct CusparseMatrixMultiplyOp<float>; variable
385 template struct CusparseMatrixMultiplyOp<double>; variable
395 struct CusparseMatrixMultiplyOp { struct
396 csrOutput operator()( in operator ()()
407 template<> struct CusparseMatrixMultiplyOp<double> { struct
[all …]