Searched refs:isDim (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 227 bool isDim(unsigned b, Dim d) const { return isDim(tensor(b), index(b), d); } in isDim() function
235 bool isDim(unsigned t, unsigned i, Dim d) const { in isDim() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 165 if (!merger.isDim(tensor, idx, Dim::kUndef)) in findAffine()
291 if (merger.isDim(tensor, i, Dim::kSparse)) in computeIterationGraph()
293 if (merger.isDim(tensor, j, Dim::kUndef)) 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()
1353 if (locals[b] && merger.isDim(b, Dim::kSparse)) { in genLocals()
1383 merger.isDim(b, Dim::kDense)) { in genLocals()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/ |
| H A D | Merger.cpp | 267 if (simple[b] && !isDim(b, kSparse)) { in simplifyCond()
297 if (bits[b] && isDim(b, d)) in hasAnyDimOf()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 676 bool isDim = (container == &dims); in composeAffineMapAndOperands() local
677 auto &repls = isDim ? dimReplacements : symReplacements; in composeAffineMapAndOperands()
681 assert(isDim ? !map->isFunctionOfDim(en.index()) in composeAffineMapAndOperands()
687 repls.push_back(isDim ? getAffineDimExpr(nDims++, ctx) in composeAffineMapAndOperands()
|
| /llvm-project-15.0.7/llvm/lib/Target/AMDGPU/AsmParser/ |
| H A D | AMDGPUAsmParser.cpp | 347 bool isDim() const { return isImmTy(ImmTyDim); } in isDim() function in __anon71e405cc0111::AMDGPUOperand
|