Searched refs:extractOp (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Conversion/VectorToSPIRV/ |
| D | VectorToSPIRV.cpp | 51 matchAndRewrite(vector::ExtractOp extractOp, ArrayRef<Value> operands, in matchAndRewrite()
53 if (extractOp.getType().isa<VectorType>() || in matchAndRewrite()
54 !spirv::CompositeType::isValid(extractOp.getVectorType())) in matchAndRewrite()
57 int32_t id = extractOp.position().begin()->cast<IntegerAttr>().getInt(); in matchAndRewrite()
59 extractOp.getLoc(), adaptor.vector(), id); in matchAndRewrite()
60 rewriter.replaceOp(extractOp, newExtract); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 652 static LogicalResult foldExtractOpFromExtractChain(ExtractOp extractOp) { in foldExtractOpFromExtractChain() argument
653 if (!extractOp.vector().getDefiningOp<ExtractOp>()) in foldExtractOpFromExtractChain()
657 ExtractOp currentOp = extractOp; in foldExtractOpFromExtractChain()
665 extractOp.setOperand(currentOp.vector()); in foldExtractOpFromExtractChain()
667 OpBuilder b(extractOp.getContext()); in foldExtractOpFromExtractChain()
669 extractOp.setAttr(ExtractOp::getPositionAttrName(), in foldExtractOpFromExtractChain()
675 static LogicalResult foldExtractOpFromTranspose(ExtractOp extractOp) { in foldExtractOpFromTranspose() argument
676 auto transposeOp = extractOp.vector().getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromTranspose()
681 auto extractedPos = extractVector<int64_t>(extractOp.position()); in foldExtractOpFromTranspose()
688 MLIRContext *ctx = extractOp.getContext(); in foldExtractOpFromTranspose()
[all …]
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestVectorTransforms.cpp | 214 SmallPtrSet<Operation *, 1> extractOp({ops->extract, ops->insert}); in runOnFunction() local
216 extractOp); in runOnFunction()
275 SmallPtrSet<Operation *, 1> extractOp({ops->extract, ops->insert}); in runOnFunction() local
276 op.getResult().replaceAllUsesExcept(ops->insert.getResult(), extractOp); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 746 auto extractOp = cast<vector::ExtractOp>(op); in matchAndRewrite() local
747 auto vectorType = extractOp.getVectorType(); in matchAndRewrite()
748 auto resultType = extractOp.getResult().getType(); in matchAndRewrite()
750 auto positionArrayAttr = extractOp.position(); in matchAndRewrite()
|