Lines Matching refs:shapeCastOp
1350 auto shapeCastOp = extractOp.getVector().getDefiningOp<vector::ShapeCastOp>(); in foldExtractFromShapeCast() local
1351 if (!shapeCastOp) in foldExtractFromShapeCast()
1361 if (destinationRank > shapeCastOp.getSourceVectorType().getRank()) in foldExtractFromShapeCast()
1369 if (getDimReverse(shapeCastOp.getSourceVectorType(), i) != in foldExtractFromShapeCast()
1390 shapeCastOp.getSourceVectorType().getRank() - destinationRank; in foldExtractFromShapeCast()
1395 getDimReverse(shapeCastOp.getSourceVectorType(), i + destinationRank); in foldExtractFromShapeCast()
1403 extractOp.setOperand(shapeCastOp.getSource()); in foldExtractFromShapeCast()
4365 LogicalResult matchAndRewrite(ShapeCastOp shapeCastOp, in matchAndRewrite() argument
4368 shapeCastOp.getSource().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
4376 DenseElementsAttr::get(shapeCastOp.getType().cast<VectorType>(), in matchAndRewrite()
4378 rewriter.replaceOpWithNewOp<arith::ConstantOp>(shapeCastOp, newAttr); in matchAndRewrite()
4391 LogicalResult matchAndRewrite(ShapeCastOp shapeCastOp, in matchAndRewrite() argument
4394 shapeCastOp.getSource().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
4403 auto shapeCastTargetShape = shapeCastOp.getResultVectorType().getShape(); in matchAndRewrite()
4412 shapeCastOp, shapeCastOp.getResultVectorType(), in matchAndRewrite()