Home
last modified time | relevance | path

Searched defs:dimOp (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DValueBoundsOpInterfaceImpl.cpp51 auto dimOp = cast<DimOp>(op); in populateBoundsForIndexValue() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DValueBoundsOpInterfaceImpl.cpp38 auto dimOp = cast<DimOp>(op); in populateBoundsForIndexValue() local
H A DTensorOps.cpp1607 if (auto dimOp = element.getDefiningOp<tensor::DimOp>()) { fold() local
[all...]
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DRuntimeOpVerification.cpp153 auto dimOp = builder.createOrFold<memref::DimOp>(loc, memref, i); in generateRuntimeVerification() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp250 auto dimOp = cast<tensor::DimOp>(op); bufferize() local
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp1137 auto dimOp = dyn_cast<AffineDimExpr>(binOp.getLHS()); in isBlockSparsity() local
1160 } else if (auto dimOp = dyn_cast<AffineDimExpr>(result)) { isBlockSparsity() local
/llvm-project/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp318 if (auto dimOp = dyn_cast<ShapedDimOpInterface>(op)) isValidDim() local
346 isDimOpValidSymbol(ShapedDimOpInterface dimOp,Region * region) isDimOpValidSymbol() argument
449 if (auto dimOp = dyn_cast<ShapedDimOpInterface>(defOp)) isValidSymbol() local
/llvm-project/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp2078 unsigned dimOp = constraints.appendDimVar(); // `op` in simplifyConstrainedMinMaxOp() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp155 Value dimOp = linalg::createOrFoldDimOp(builder, loc, mem, r); in genAllocMemRef() local