Lines Matching refs:getDefiningOp
57 if (auto c = mask.getDefiningOp<arith::ConstantOp>()) { in get1DMaskFormat()
75 } else if (auto m = mask.getDefiningOp<ConstantMaskOp>()) { in get1DMaskFormat()
164 auto indexA = transferA.indices()[i].getDefiningOp<arith::ConstantOp>(); in isDisjointTransferIndices()
165 auto indexB = transferB.indices()[i].getDefiningOp<arith::ConstantOp>(); in isDisjointTransferIndices()
917 maybeContraction.getDefiningOp()); in matchAndRewrite()
921 contractionOp.getAcc().getDefiningOp())) { in matchAndRewrite()
987 if (auto splat = getVector().getDefiningOp<vector::SplatOp>()) in fold()
1015 return pos.getDefiningOp<arith::ConstantIndexOp>().value(); in build()
1078 if (!extractOp.getVector().getDefiningOp<ExtractOp>()) in foldExtractOpFromExtractChain()
1085 while (ExtractOp nextOp = currentOp.getVector().getDefiningOp<ExtractOp>()) { in foldExtractOpFromExtractChain()
1148 nextInsertOp = v.getDefiningOp<vector::InsertOp>(); in updateStateForNextIteration()
1149 nextTransposeOp = v.getDefiningOp<vector::TransposeOp>(); in updateStateForNextIteration()
1316 Operation *defOp = extractOp.getVector().getDefiningOp(); in foldExtractFromBroadcast()
1350 auto shapeCastOp = extractOp.getVector().getDefiningOp<vector::ShapeCastOp>(); in foldExtractFromShapeCast()
1410 extractOp.getVector().getDefiningOp<vector::ExtractStridedSliceOp>(); in foldExtractFromExtractStrided()
1453 auto insertOp = op.getVector().getDefiningOp<InsertStridedSliceOp>(); in foldExtractStridedOpFromInsertChain()
1506 insertOp = insertOp.getDest().getDefiningOp<InsertStridedSliceOp>(); in foldExtractStridedOpFromInsertChain()
1538 Operation *defOp = extractOp.getVector().getDefiningOp(); in matchAndRewrite()
1570 auto constantOp = extractOp.getVector().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
1639 auto insert = getVector().getDefiningOp<vector::InsertMapOp>(); in fold()
1755 auto srcBroadcast = broadcastOp.getSource().getDefiningOp<BroadcastOp>(); in matchAndRewrite()
1889 auto v1Splat = op.getV1().getDefiningOp<SplatOp>(); in matchAndRewrite()
1890 auto v2Splat = op.getV2().getDefiningOp<SplatOp>(); in matchAndRewrite()
1973 return pos.getDefiningOp<arith::ConstantIndexOp>().value(); in build()
2033 auto srcSplat = op.getSource().getDefiningOp<SplatOp>(); in matchAndRewrite()
2034 auto dstSplat = op.getDest().getDefiningOp<SplatOp>(); in matchAndRewrite()
2240 insertStridedSliceOp.getSource().getDefiningOp<vector::SplatOp>(); in matchAndRewrite()
2242 insertStridedSliceOp.getDest().getDefiningOp<vector::SplatOp>(); in matchAndRewrite()
2266 .getDefiningOp<vector::ExtractStridedSliceOp>(); in matchAndRewrite()
2434 return isa_and_nonnull<arith::ConstantIndexOp>(operand.getDefiningOp()); in verify()
2441 cast<arith::ConstantIndexOp>(operand.getDefiningOp()).value(); in verify()
2445 cast<arith::ConstantIndexOp>(operand.getDefiningOp()).value(); in verify()
2546 auto insertOp = op.getVector().getDefiningOp<InsertStridedSliceOp>(); in foldExtractStridedOpFromInsertChain()
2591 insertOp = insertOp.getDest().getDefiningOp<InsertStridedSliceOp>(); in foldExtractStridedOpFromInsertChain()
2626 auto *defOp = extractStridedSliceOp.getVector().getDefiningOp(); in matchAndRewrite()
2680 extractStridedSliceOp.getVector().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
2703 auto broadcast = op.getVector().getDefiningOp<BroadcastOp>(); in matchAndRewrite()
2746 auto splat = op.getVector().getDefiningOp<SplatOp>(); in matchAndRewrite()
3072 auto castOp = operand.get().getDefiningOp<memref::CastOp>(); in foldMemRefCast()
3084 auto castOp = operand.get().getDefiningOp<tensor::CastOp>(); in foldTensorCast()
3100 auto cstOp = index.getDefiningOp<arith::ConstantIndexOp>(); in isInBounds()
3158 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in foldRAW()
3166 defWrite = defWrite.getSource().getDefiningOp<vector::TransferWriteOp>(); in foldRAW()
3228 auto extractOp = xferOp.getSource().getDefiningOp<tensor::ExtractSliceOp>(); in matchAndRewrite()
3319 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in matchAndRewrite()
3537 auto read = write.getVector().getDefiningOp<vector::TransferReadOp>(); in foldReadInitWrite()
3564 auto cstOp = v.getDefiningOp<arith::ConstantIndexOp>(); in foldReadInitWrite()
3601 auto read = write.getVector().getDefiningOp<vector::TransferReadOp>(); in foldWAR()
3669 writeOp.getSource().getDefiningOp<vector::TransferWriteOp>(); 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()
4329 if (auto otherOp = getSource().getDefiningOp<ShapeCastOp>()) { in fold()
4351 if (auto bcastOp = getSource().getDefiningOp<BroadcastOp>()) { in fold()
4368 shapeCastOp.getSource().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
4394 shapeCastOp.getSource().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
4463 if (auto otherOp = getSource().getDefiningOp<BitCastOp>()) { in fold()
4641 transposeOp.getVector().getDefiningOp<vector::TransposeOp>(); in matchAndRewrite()
4663 auto bcastOp = transposeOp.getVector().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
4685 auto splatOp = transposeOp.getVector().getDefiningOp<vector::SplatOp>(); in matchAndRewrite()
4787 return !isa_and_nonnull<arith::ConstantIndexOp>(operand.getDefiningOp()); in matchAndRewrite()
4808 auto *defOp = std::get<0>(it).getDefiningOp(); in matchAndRewrite()