Searched refs:fusedOp (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 91 Operation *fusedOp, LinalgOp producer, in generateFusedTensorOpRegion() argument
99 fusedOp->getRegion(0).push_back(fusedBlock); in generateFusedTensorOpRegion()
217 LinalgOp fusedOp; in fuseTensorOpsImpl() local
220 fusedOp = in fuseTensorOpsImpl()
233 fusedOp = rewriter in fuseTensorOpsImpl()
261 generateFusedTensorOpRegion(rewriter, fusedOp.getOperation(), producer, in fuseTensorOpsImpl()
264 return SmallVector<Value, 1>(fusedOp->getResults()); in fuseTensorOpsImpl()
584 LinalgOp fusedOp = createLinalgOpOfSameType( in fuseWithReshapeByExpansion() local
589 Region &fusedRegion = fusedOp->getRegion(0); in fuseWithReshapeByExpansion()
616 edsc::ScopedContext scopedContext(rewriter, fusedOp.getLoc()); in fuseWithReshapeByExpansion()
[all …]
|
| D | Fusion.cpp | 827 LinalgOp fusedOp = fuse(builder, candidate.value(), fusedLoopsAndRanges); in fuseOperations() local
828 fusedOps[fusionCandidates.size() - candidate.index() - 1] = fusedOp; in fuseOperations()
829 builder.setInsertionPoint(fusedOp); in fuseOperations()
|
| D | Transforms.cpp | 229 for (auto fusedOp : tiledAndFusedOps->fusedProducers) { in matchAndRewrite() local
230 fusedOpMarker.replaceLinalgMarker(rewriter, fusedOp.getOperation()); in matchAndRewrite()
|