Searched refs:kSparse (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 311 merger.setDim(t0, l0, Dim::kSparse); in MergerTest3T1L()
315 merger.setDim(t1, l0, Dim::kSparse); in MergerTest3T1L()
334 merger.setDim(t0, l0, Dim::kSparse); in MergerTest4T1L()
338 merger.setDim(t1, l0, Dim::kSparse); in MergerTest4T1L()
342 merger.setDim(t2, l0, Dim::kSparse); in MergerTest4T1L()
365 merger.setDim(t0, l0, Dim::kSparse); in MergerTest3T1LD()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 149 return Dim::kSparse; in toDim()
291 if (merger.isDim(tensor, i, Dim::kSparse)) in computeIterationGraph()
339 if (merger.isDim(tensor, i, Dim::kSparse)) { in isAdmissableTensorExp()
515 if (merger.isDim(tensor, idx, Dim::kSparse)) { in genBuffers()
1095 if (merger.isDim(b, Dim::kSparse)) { in genInit()
1199 bool isSparse = merger.isDim(fb, Dim::kSparse); in genFor()
1272 if (indices[b] && merger.isDim(b, Dim::kSparse)) { in genWhile()
1305 if (indices[b] && merger.isDim(b, Dim::kSparse)) { in genWhile()
1353 if (locals[b] && merger.isDim(b, Dim::kSparse)) { in genLocals()
1441 if (induction[b] && merger.isDim(b, Dim::kSparse)) { in genWhileInduction()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 24 enum Dim { kSparse, kDense, kSingle, kUndef }; enumerator
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/ |
| H A D | Merger.cpp | 265 bool reset = isSingleton && hasAnyDimOf(simple, kSparse); in simplifyCond()
267 if (simple[b] && !isDim(b, kSparse)) { in simplifyCond()
292 return !hasAnyDimOf(tmp, kSparse); in onlyDenseDiff()
588 case kSparse: in dumpBits()
|