Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp790 LogicalResult matchAndRewrite(vector::ShapeCastOp shapeCastOp, in matchAndRewrite()
794 shapeCastOp.source().getType().dyn_cast_or_null<TupleType>(); in matchAndRewrite()
796 shapeCastOp.result().getType().dyn_cast_or_null<TupleType>(); in matchAndRewrite()
802 Location loc = shapeCastOp.getLoc(); in matchAndRewrite()
807 loc, sourceTupleType.getType(i), shapeCastOp.source(), in matchAndRewrite()
814 rewriter.replaceOpWithNewOp<vector::TupleOp>(shapeCastOp, resultTupleType, in matchAndRewrite()
906 } else if (auto shapeCastOp = dyn_cast<vector::ShapeCastOp>(op)) { in getProducerValue() local
907 if (shapeCastOp.source().getType().isa<TupleType>()) in getProducerValue()
910 auto sourceVectorType = shapeCastOp.getSourceVectorType(); in getProducerValue()
913 auto resultVectorType = shapeCastOp.getResultVectorType(); in getProducerValue()
[all …]
DVectorOps.cpp847 auto shapeCastOp = extractOp.vector().getDefiningOp<vector::ShapeCastOp>(); in foldExtractFromShapeCast() local
848 if (!shapeCastOp) in foldExtractFromShapeCast()
858 if (destinationRank > shapeCastOp.getSourceVectorType().getRank()) in foldExtractFromShapeCast()
866 if (getDimReverse(shapeCastOp.getSourceVectorType(), i) != in foldExtractFromShapeCast()
887 shapeCastOp.getSourceVectorType().getRank() - destinationRank; in foldExtractFromShapeCast()
892 getDimReverse(shapeCastOp.getSourceVectorType(), i + destinationRank); in foldExtractFromShapeCast()
900 extractOp.setOperand(shapeCastOp.source()); in foldExtractFromShapeCast()
2646 LogicalResult matchAndRewrite(ShapeCastOp shapeCastOp, in matchAndRewrite() argument
2648 auto constantOp = shapeCastOp.source().getDefiningOp<ConstantOp>(); in matchAndRewrite()
2656 shapeCastOp.getType().cast<VectorType>(), dense.getSplatValue()); in matchAndRewrite()
[all …]