Home
last modified time | relevance | path

Searched refs:fusedBlock (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp162 Block *fusedBlock = new Block(); in matchAndRewrite() local
163 fusedOp.region().push_back(fusedBlock); in matchAndRewrite()
166 addArg(mapper, fusedBlock, prodBlock.getArgument(i)); in matchAndRewrite()
167 addArg(mapper, fusedBlock, consBlock.getArgument(1 - other)); in matchAndRewrite()
168 addArg(mapper, fusedBlock, prodBlock.getArgument(num - 1)); in matchAndRewrite()
172 rewriter.setInsertionPointToStart(fusedBlock); in matchAndRewrite()
179 mapper.map(consBlock.getArgument(other), fusedBlock->back().getResult(0)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp148 Block *fusedBlock = new Block(); in generateFusedElementwiseOpRegion() local
149 fusedOp.region().push_back(fusedBlock); in generateFusedElementwiseOpRegion()
152 rewriter.setInsertionPointToStart(fusedBlock); in generateFusedElementwiseOpRegion()
180 mapper.map(bbArg, fusedBlock->addArgument(bbArg.getType(), bbArg.getLoc())); in generateFusedElementwiseOpRegion()
188 mapper.map(bbArg, fusedBlock->addArgument(bbArg.getType(), bbArg.getLoc())); in generateFusedElementwiseOpRegion()
198 mapper.map(bbArg, fusedBlock->addArgument(bbArg.getType(), bbArg.getLoc())); in generateFusedElementwiseOpRegion()
205 mapper.map(bbArg, fusedBlock->addArgument(bbArg.getType(), bbArg.getLoc())); in generateFusedElementwiseOpRegion()
209 mapper.map(bbArg, fusedBlock->addArgument(bbArg.getType(), bbArg.getLoc())); in generateFusedElementwiseOpRegion()
245 assert(fusedBlock->getNumArguments() == fusedOp.getNumOperands() && in generateFusedElementwiseOpRegion()