Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp599 Block *fusedEntryBlock = new Block; in fuseWithReshapeByExpansion() local
600 fusedRegion.push_back(fusedEntryBlock); in fuseWithReshapeByExpansion()
615 rewriter.setInsertionPointToStart(fusedEntryBlock); in fuseWithReshapeByExpansion()
619 Value linearizedIndex = fusedEntryBlock->addArgument(indexType); in fuseWithReshapeByExpansion()
628 linearizedIndex + fusedEntryBlock->addArgument(indexType); in fuseWithReshapeByExpansion()
635 fusedEntryBlock->addArgument(originalRegion.getArgument(i).getType()); in fuseWithReshapeByExpansion()
637 rewriter.mergeBlocks(fusedEntryBlock->getNextNode(), fusedEntryBlock, in fuseWithReshapeByExpansion()