Searched refs:kTensor (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 225 case kTensor: in compareExpression()
310 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest3T1L()
314 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest3T1L()
318 merger.addExp(Kind::kTensor, t2, -1u); in MergerTest3T1L()
333 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest4T1L()
337 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest4T1L()
341 merger.addExp(Kind::kTensor, t2, -1u); in MergerTest4T1L()
345 merger.addExp(Kind::kTensor, t3, -1u); in MergerTest4T1L()
364 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest3T1LD()
368 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest3T1LD()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/ |
| H A D | Merger.cpp | 29 case kTensor: in TensorExp()
233 if (tensorExps[e].kind == kTensor && tensorExps[e].tensor == outTensor) in optimizeSet()
305 case kTensor: in isSingleCondition()
392 case kTensor: in kindToOpSymbol()
484 case kTensor: in dumpExp()
615 case kTensor: in buildLattices()
625 if (kind == kTensor) { in buildLattices()
877 return addExp(kTensor, argN); in buildTensorExp()
1073 case kTensor: in buildExp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 29 kTensor = 0, enumerator
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 953 if (merger.exp(exp).kind == Kind::kTensor) in genExp()
995 bool atStart, Kind last = Kind::kTensor) { in genInvariants()
998 if (merger.exp(exp).kind == Kind::kTensor) { in genInvariants()
|