Searched refs:setLevelAndType (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 318 merger.setLevelAndType(tid(0), lid(0), 0, LevelFormat::Compressed); in MergerTest3T1L()
320 merger.setLevelAndType(tid(1), lid(0), 0, LevelFormat::Compressed); in MergerTest3T1L()
322 merger.setLevelAndType(tid(2), lid(0), 0, GetParam()); in MergerTest3T1L()
336 merger.setLevelAndType(tid(0), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
338 merger.setLevelAndType(tid(1), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
340 merger.setLevelAndType(tid(2), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
342 merger.setLevelAndType(tid(3), lid(0), 0, GetParam()); in MergerTest4T1L()
360 merger.setLevelAndType(tid(0), lid(0), 0, LevelFormat::Compressed); in MergerTest3T1LD()
362 merger.setLevelAndType(tid(1), lid(0), 0, GetParam()); in MergerTest3T1LD()
364 merger.setLevelAndType(tid(2), lid(0), 0, GetParam()); in MergerTest3T1LD()
[all …]
|
| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 426 void setLevelAndType(TensorId t, LoopId i, Level lvl, LevelType lt) { in setLevelAndType() function
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 83 merger.setLevelAndType(tid, idx, lvl, lt); in findAffine()
137 merger.setLevelAndType(tensor, idx, lvl, lt); in findDepIdxSet()
|