Home
last modified time | relevance | path

Searched defs:indexOp (Results 1 – 11 of 11) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DValueBoundsOpInterfaceImpl.cpp24 auto indexOp = cast<IndexOp>(op); in populateBoundsForIndexValue() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DInterchange.cpp90 for (IndexOp indexOp : interchangeGenericOp() local
H A DLoops.cpp203 for (IndexOp indexOp : llvm::make_early_inc_range(r->getOps<IndexOp>())) in replaceIndexOpsByInductionVariables() local
H A DTilingInterfaceImpl.cpp54 if (auto indexOp = dyn_cast<IndexOp>(&op)) { in inlinePayload() local
H A DElementwiseOpFusion.cpp196 for (IndexOp indexOp : generateFusedElementwiseOpRegion() local
713 for (IndexOp indexOp : updateExpandedGenericOpRegion() local
1537 for (auto indexOp : indexOps) { generateCollapsedIndexingRegion() local
[all...]
H A DDropUnitDims.cpp236 if (unitDims.count(indexOp.getDim()) != 0) { in replaceUnitDimIndexOps() local
H A DVectorization.cpp692 IndexOp indexOp = dyn_cast<linalg::IndexOp>(op); vectorizeLinalgIndex() local
824 if (auto indexOp = dyn_cast<linalg::IndexOp>(defOp)) isLoopInvariantIdx() local
885 if (auto indexOp = dyn_cast<linalg::IndexOp>(defOp)) { isContiguousLoadIdx() local
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DKernelOutlining.cpp69 for (const auto &indexOp : enumerate(indexOps)) in injectGpuIndexOperations() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp1314 if (auto indexOp = dyn_cast<linalg::IndexOp>(def)) in buildTensorExp() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp601 if (auto indexOp = dyn_cast<linalg::IndexOp>(def)) relinkBranch() local
/llvm-project/flang/lib/Optimizer/HLFIR/IR/
H A DHLFIROps.cpp1858 canonicalize(hlfir::ForallIndexOp indexOp,mlir::PatternRewriter & rewriter) canonicalize() argument