Lines Matching refs:getSource

350     return getSource();  in fold()
1232 res = nextInsertOp.getSource(); in handleInsertOpWithMatchingPos()
1252 res = nextInsertOp.getSource(); in handleInsertOpWithPrefixPos()
1403 extractOp.setOperand(shapeCastOp.getSource()); in foldExtractFromShapeCast()
1497 op.getVectorMutable().assign(insertOp.getSource()); in foldExtractStridedOpFromInsertChain()
1736 return getSource(); in fold()
1755 auto srcBroadcast = broadcastOp.getSource().getDefiningOp<BroadcastOp>(); in matchAndRewrite()
1759 broadcastOp, broadcastOp.getVectorType(), srcBroadcast.getSource()); in matchAndRewrite()
2021 insertOp, insertOp.getDestVectorType(), insertOp.getSource()); in matchAndRewrite()
2033 auto srcSplat = op.getSource().getDefiningOp<SplatOp>(); in matchAndRewrite()
2059 return getSource(); in fold()
2240 insertStridedSliceOp.getSource().getDefiningOp<vector::SplatOp>(); in matchAndRewrite()
2265 insertStridedSliceOp.getSource() in matchAndRewrite()
2295 return getSource(); in fold()
2581 op.setOperand(insertOp.getSource()); in foldExtractStridedOpFromInsertChain()
2706 auto srcVecType = broadcast.getSource().getType().dyn_cast<VectorType>(); in matchAndRewrite()
2721 Value source = broadcast.getSource(); in matchAndRewrite()
2957 p << " " << getSource() << "[" << getIndices() << "], " << getPadding(); in print()
3158 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in foldRAW()
3166 defWrite = defWrite.getSource().getDefiningOp<vector::TransferWriteOp>(); in foldRAW()
3192 effects.emplace_back(MemoryEffects::Read::get(), getSource(), in getEffects()
3228 auto extractOp = xferOp.getSource().getDefiningOp<tensor::ExtractSliceOp>(); in matchAndRewrite()
3281 xferOp, xferOp.getVectorType(), extractOp.getSource(), newIndices, in matchAndRewrite()
3319 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in matchAndRewrite()
3477 p << " " << getVector() << ", " << getSource() << "[" << getIndices() << "]"; in print()
3532 write.getSource().getType().dyn_cast<RankedTensorType>(); in foldReadInitWrite()
3554 if (read.getSource().getType() != rankedTensorType) in foldReadInitWrite()
3571 results.push_back(read.getSource()); in foldReadInitWrite()
3577 return read.getSource() == write.getSource() && in checkSameValueWAR()
3599 if (!write.getSource().getType().isa<RankedTensorType>()) in foldWAR()
3607 results.push_back(read.getSource()); in foldWAR()
3630 effects.emplace_back(MemoryEffects::Write::get(), getSource(), in getEffects()
3669 writeOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in matchAndRewrite()
3672 writeToModify.getSourceMutable().assign(defWrite.getSource()); in matchAndRewrite()
3684 defWrite = defWrite.getSource().getDefiningOp<vector::TransferWriteOp>(); in matchAndRewrite()
3714 auto xferOp = insertOp.getSource().getDefiningOp<TransferWriteOp>(); in matchAndRewrite()
3808 insertOp.getSource().getDefiningOp<tensor::ExtractSliceOp>(); in matchAndRewrite()
3811 auto transferOp = extractOp.getSource().getDefiningOp<TransferWriteOp>(); in matchAndRewrite()
4313 auto sourceVectorType = getSource().getType().dyn_cast_or_null<VectorType>(); in verify()
4325 if (getSource().getType() == getResult().getType()) in fold()
4326 return getSource(); in fold()
4329 if (auto otherOp = getSource().getDefiningOp<ShapeCastOp>()) { in fold()
4330 if (getResult().getType() == otherOp.getSource().getType()) in fold()
4331 return otherOp.getSource(); in fold()
4334 VectorType srcType = otherOp.getSource().getType().cast<VectorType>(); in fold()
4346 setOperand(otherOp.getSource()); in fold()
4351 if (auto bcastOp = getSource().getDefiningOp<BroadcastOp>()) { in fold()
4353 return bcastOp.getSource(); in fold()
4368 shapeCastOp.getSource().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
4394 shapeCastOp.getSource().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
4413 broadcastOp.getSource()); in matchAndRewrite()
4459 if (getSource().getType() == getResult().getType()) in fold()
4460 return getSource(); in fold()
4463 if (auto otherOp = getSource().getDefiningOp<BitCastOp>()) { in fold()
4464 if (getResult().getType() == otherOp.getSource().getType()) in fold()
4465 return otherOp.getSource(); in fold()
4467 setOperand(otherOp.getSource()); in fold()
4670 transposeOp, transposeOp.getResultType(), bcastOp.getSource()); in matchAndRewrite()