Searched defs:aTp (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseGPUCodegen.cpp | 411 static bool isAdmissibleMetaData(SparseTensorType &aTp) { in isAdmissibleMetaData() argument
417 static bool isAdmissibleCOO(SparseTensorType &aTp) { in isAdmissibleCOO() argument
425 isAdmissibleCSR(SparseTensorType & aTp) isAdmissibleCSR() argument
432 isAdmissibleCSC(SparseTensorType & aTp) isAdmissibleCSC() argument
439 isAdmissibleBSR(SparseTensorType & aTp) isAdmissibleBSR() argument
452 isAdmissible24(SparseTensorType & aTp) isAdmissible24() argument
462 SparseTensorType aTp = getSparseTensorType(a); isConversionInto24() local
470 getCuSparseFormat(SparseTensorType aTp,SparseTensorType bTp,SparseTensorType cTp,bool enableRT,bool isMatVec) getCuSparseFormat() argument
518 genSpMat(OpBuilder & builder,Location loc,SparseTensorType & aTp,Type handleTp,Type tokenTp,Value token,Value sz1,Value sz2,Value nseA,Value rowA,Value colA,Value valA,CuSparseFormat format,bool enableRT) genSpMat() argument
571 SparseTensorType aTp = getSparseTensorType(a); rewriteSpMV() local
668 SparseTensorType aTp = getSparseTensorType(a); rewriteSpMM() local
770 SparseTensorType aTp = getSparseTensorType(a); rewriteSpGEMM() local
1061 SparseTensorType aTp = getSparseTensorType(a); rewriteSDDMM() local
[all...] |