Lines Matching refs:getSource
100 castOp.getSource().getType()); in canFoldIntoConsumerOp()
126 return preservesStaticInformation(castOp.getSource().getType(), in canFoldIntoProducerOp()
253 tensorCast.getType().getShape() == tensorCast.getSource() in matchAndRewrite()
275 extractOperand.getSource(), extractOperand.getMixedOffsets(), sizes, in matchAndRewrite()
312 auto type = getSource().getType(); in verify()
331 auto tensorType = getSource().getType().dyn_cast<RankedTensorType>(); in fold()
341 Operation *definingOp = getSource().getDefiningOp(); in fold()
386 auto castOp = dimOp.getSource().getDefiningOp<CastOp>(); in matchAndRewrite()
722 extract, tensorCast.getSource(), extract.getIndices()); in matchAndRewrite()
761 TensorType operandType = getSource().getType().cast<TensorType>(); in verify()
1200 auto castOp = sliceOp.getSource().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1214 sliceOp.getLoc(), resultType, castOp.getSource(), sliceOp.getOffsets(), in matchAndRewrite()
1270 if (!matchPattern(op.getSource(), m_Constant(&attr))) in matchAndRewrite()
1278 auto sourceType = op.getSource().getType().cast<ShapedType>(); in matchAndRewrite()
1407 auto insertOp = extractOp.getSource().getDefiningOp<InsertSliceOp>(); in foldExtractAfterInsertSlice()
1410 if (insertOp && insertOp.getSource().getType() == extractOp.getType() && in foldExtractAfterInsertSlice()
1412 return insertOp.getSource(); in foldExtractAfterInsertSlice()
1425 return this->getSource(); in fold()
1540 prevInsertOp.getSource().getType() != insertOp.getSource().getType() || in foldInsertAfterInsertSlice()
1559 return static_cast<OpFoldResult>(insertOp.getSource()); in foldInsertOp()
1618 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()
1671 return castOp.getSource(); in matchAndRewrite()
1674 getSourceOfCastOp(insertSliceOp.getSource()); in matchAndRewrite()
1680 (sourceCastSource ? *sourceCastSource : insertSliceOp.getSource()); in matchAndRewrite()
1765 insertSliceOp.getLoc(), newSrcType, insertSliceOp.getSource()); in matchAndRewrite()
1824 auto sourceType = getSource().getType().cast<RankedTensorType>(); in verify()
1966 padTensorOp.getSource()); in matchAndRewrite()
1977 auto castOp = padTensorOp.getSource().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1982 castOp.getSource().getType().cast<RankedTensorType>(), in matchAndRewrite()
1989 padTensorOp.getSourceMutable().assign(castOp.getSource()); in matchAndRewrite()
1993 padTensorOp->getLoc(), newResultType, padTensorOp.getSource(), in matchAndRewrite()
2026 padTensorOp.getSource(), padTensorOp.getLow(), padTensorOp.getHigh(), in matchAndRewrite()
2076 auto innerSliceOp = padOp.getSource().getDefiningOp<ExtractSliceOp>(); in matchAndRewrite()
2079 auto outerPadOp = innerSliceOp.getSource().getDefiningOp<PadOp>(); in matchAndRewrite()
2082 auto outerSliceOp = outerPadOp.getSource().getDefiningOp<ExtractSliceOp>(); in matchAndRewrite()
2181 padOp.getLoc(), outerSliceOp.getSource(), newOffsets, newSizes, in matchAndRewrite()
2228 return getSource(); in fold()
2326 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()