Home
last modified time | relevance | path

Searched refs:transposeOp (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorMultiDimReductionTransforms.cpp81 auto transposeOp = rewriter.create<vector::TransposeOp>(loc, src, indices); in matchAndRewrite() local
90 multiReductionOp, transposeOp.getResult(), multiReductionOp.getAcc(), in matchAndRewrite()
H A DVectorTransforms.cpp1046 if (!transposeOp) in matchAndRewrite()
1049 transposeOp.getTransp(perm); in matchAndRewrite()
1051 extractVector<unsigned>(transposeOp.getTransp()), in matchAndRewrite()
1054 *operand = transposeOp.getVector(); in matchAndRewrite()
1259 auto transposeOp = operand.getDefiningOp<vector::TransposeOp>(); in matchAndRewrite() local
1260 if (transposeOp) { in matchAndRewrite()
1261 transposeMaps.push_back(transposeOp.getTransp()); in matchAndRewrite()
1262 srcType = transposeOp.getVectorType(); in matchAndRewrite()
1286 auto transposeOp = operand.getDefiningOp<vector::TransposeOp>(); in matchAndRewrite() local
1287 if (transposeOp) { in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1812 matchAndRewrite(memref::TransposeOp transposeOp, OpAdaptor adaptor, in matchAndRewrite() argument
1814 auto loc = transposeOp.getLoc(); in matchAndRewrite()
1818 if (transposeOp.getPermutation().isIdentity()) in matchAndRewrite()
1819 return rewriter.replaceOp(transposeOp, {viewMemRef}), success(); in matchAndRewrite()
1822 rewriter, loc, typeConverter->convertType(transposeOp.getShapedType())); in matchAndRewrite()
1836 llvm::enumerate(transposeOp.getPermutation().getResults())) { in matchAndRewrite()
1845 rewriter.replaceOp(transposeOp, {targetMemRef}); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp459 auto transposeOp = b.create<GenericOp>( in makeTransposeOp() local
464 Region &body = transposeOp.getRegion(); in makeTransposeOp()
471 b.create<YieldOp>(loc, transposeOp.getRegion().front().getArgument(0)); in makeTransposeOp()
472 return transposeOp; in makeTransposeOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp4621 LogicalResult matchAndRewrite(vector::TransposeOp transposeOp, in matchAndRewrite() argument
4641 transposeOp.getVector().getDefiningOp<vector::TransposeOp>(); in matchAndRewrite()
4646 getPermutation(parentTransposeOp), getPermutation(transposeOp)); in matchAndRewrite()
4649 transposeOp, transposeOp.getResult().getType(), in matchAndRewrite()
4661 LogicalResult matchAndRewrite(vector::TransposeOp transposeOp, in matchAndRewrite()
4663 auto bcastOp = transposeOp.getVector().getDefiningOp<vector::BroadcastOp>(); in matchAndRewrite()
4670 transposeOp, transposeOp.getResultType(), bcastOp.getSource()); in matchAndRewrite()
4683 LogicalResult matchAndRewrite(TransposeOp transposeOp, in matchAndRewrite() argument
4685 auto splatOp = transposeOp.getVector().getDefiningOp<vector::SplatOp>(); in matchAndRewrite()
4690 transposeOp, transposeOp.getResultType(), splatOp.getInput()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp447 for (GenericOp transposeOp : transposeOps) in returningMatchAndRewrite() local
448 filter.replaceLinalgTransformationFilter(rewriter, transposeOp); in returningMatchAndRewrite()
H A DVectorization.cpp298 auto transposeOp = in vectorizeLinalgIndex() local
300 return VectorizationResult{VectorizationStatus::NewOp, transposeOp}; in vectorizeLinalgIndex()