Lines Matching defs:candidateSliceOp
1139 RewriterBase &rewriter, tensor::ExtractSliceOp candidateSliceOp,
1144 getUntiledProducerFromSliceSource(&candidateSliceOp.getSourceMutable(),
1151 rewriter.setInsertionPoint(candidateSliceOp);
1169 clonedOpDestinationTensors[resultNumber] = candidateSliceOp.getSource();
1178 llvm::to_vector(candidateSliceOp->getOperands());
1181 mlir::clone(rewriter, candidateSliceOp,
1182 candidateSliceOp->getResultTypes(), candidateSliceOpOperands);
1191 // Note: Do not delete the candidateSliceOp, since its passed in from the
1193 rewriter.replaceAllUsesWith(candidateSliceOp,
1662 checkAssumptionForFusingConsumer(tensor::InsertSliceOp candidateSliceOp) {
1663 Value result = candidateSliceOp.getResult();
1897 tensor::InsertSliceOp candidateSliceOp) {
1898 if (failed(checkAssumptionForFusingConsumer(candidateSliceOp)))
1900 Value sliceResult = candidateSliceOp.getResult();
1905 Operation *containingOp = candidateSliceOp->getParentOp();
1918 tensor::ParallelInsertSliceOp candidateSliceOp) {
1920 Value sliceDest = candidateSliceOp.getDest();
1925 if (containingOp != candidateSliceOp->getParentOp()->getParentOp())
1956 Operation *candidateSliceOp) {
1958 candidateSliceOp))
1961 bool isInsertSliceOp = isa<tensor::InsertSliceOp>(candidateSliceOp);
1966 getUntiledConsumerFromSlice(rewriter, candidateSliceOp);
1968 return rewriter.notifyMatchFailure(candidateSliceOp,
1987 candidateSliceOp->getParentOfType<LoopLikeOpInterface>();
2038 // candidateSliceOp whereas in the scf.forall case this is created from the
2042 dyn_cast<tensor::ParallelInsertSliceOp>(candidateSliceOp)) {
2049 rewriter.setInsertionPoint(candidateSliceOp);
2051 cast<tensor::InsertSliceOp>(rewriter.clone(*candidateSliceOp));
2097 candidateSliceOp, "containingOp's result yield with stride");