Home
last modified time | relevance | path

Searched refs:shapeCastOp (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1351 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()
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()
4391 LogicalResult matchAndRewrite(ShapeCastOp shapeCastOp, in matchAndRewrite() argument
4394 shapeCastOp.getSource().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp221 LogicalResult matchAndRewrite(vector::ShapeCastOp shapeCastOp, in matchAndRewrite()
225 shapeCastOp.getSource().getType().dyn_cast_or_null<VectorType>(); in matchAndRewrite()
227 shapeCastOp.getResult().getType().dyn_cast_or_null<VectorType>(); in matchAndRewrite()
233 shapeCastOp.getSource().getDefiningOp()); in matchAndRewrite()
245 rewriter.replaceOp(shapeCastOp, sourceShapeCastOp.getSource()); in matchAndRewrite()