Home
last modified time | relevance | path

Searched refs:blockArgument (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestAffineLoopParametricTiling.cpp58 for (BlockArgument blockArgument : in getTilingParameters() local
60 if (blockArgument.getArgNumber() < nestDepth) { in getTilingParameters()
61 assert(blockArgument.getType().isIndex() && in getTilingParameters()
63 tilingParameters.push_back(blockArgument); in getTilingParameters()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDetensorize.cpp110 for (BlockArgument blockArgument : block.getArguments()) { in matchAndRewrite() local
111 int idx = blockArgument.getArgNumber(); in matchAndRewrite()
113 if (blockArgsToDetensor.count(blockArgument)) in matchAndRewrite()
458 for (BlockArgument blockArgument : block.getArguments()) in compute() local
459 blockArgsToDetensor.insert(blockArgument); in compute()
499 blockArgsToDetensor, [&](BlockArgument blockArgument) { in runOnOperation() argument
500 return blockArgument.getOwner() == &block && in runOnOperation()
501 !typeConverter.isLegal(blockArgument.getType()); in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgInterfaces.td704 Return the operand for a `blockArgument`.
708 /*args=*/(ins "BlockArgument":$blockArgument),
711 assert(blockArgument.getOwner() == getBlock());
713 blockArgument.getArgNumber());