Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp98 Block *fusedBlock = new Block(); in generateFusedTensorOpRegion() local
99 fusedOp->getRegion(0).push_back(fusedBlock); in generateFusedTensorOpRegion()
102 rewriter.setInsertionPointToStart(fusedBlock); in generateFusedTensorOpRegion()
126 fusedBlock->addArgument(rewriter.getIndexType()); in generateFusedTensorOpRegion()
139 fusedBlock->getArguments().take_front(numFusedOpIndices)); in generateFusedTensorOpRegion()
143 fusedBlock->addArgument(producerArg.value().getType())); in generateFusedTensorOpRegion()
154 fusedBlock->getArgument(consumerArg.index())); in generateFusedTensorOpRegion()
157 fusedBlock->addArgument(consumerArg.value().getType())); in generateFusedTensorOpRegion()