| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/Utils/ |
| H A D | VectorUtils.h | 36 Value createOrFoldDimOp(OpBuilder &b, Location loc, Value source, int64_t dim);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 481 static Value createOrFoldDimOp(OpBuilder &b, Location loc, Value source, in createOrFoldDimOp() function
495 res.push_back(createOrFoldDimOp(b, loc, opOperand->get(), i)); in createFlatListOfOperandDims()
603 shapes.push_back(createOrFoldDimOp(b, loc, opOperand->get(), dim)); in reifyResultShapes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/ |
| H A D | VectorUtils.cpp | 37 Value mlir::vector::createOrFoldDimOp(OpBuilder &b, Location loc, Value source, in createOrFoldDimOp() function in mlir::vector
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferSplitRewritePatterns.cpp | 81 b, sum, vector::createOrFoldDimOp(b, loc, xferOp.source(), indicesIdx)); in createInBoundsCond()
203 Value dimMemRef = vector::createOrFoldDimOp(b, xferOp.getLoc(), in createSubViewIntersection()
|
| H A D | VectorTransforms.cpp | 2543 vector::createOrFoldDimOp(rewriter, loc, xferOp.getSource(), lastIndex); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 50 Value createOrFoldDimOp(OpBuilder &b, Location loc, Value source, int64_t dim);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 195 Value createOrFoldDimOp(OpBuilder &b, Location loc, Value source, int64_t dim) { in createOrFoldDimOp() function
210 dynOperands.push_back(createOrFoldDimOp(b, loc, val, dim.index())); in getDynOperands()
817 Value dim = createOrFoldDimOp(builder, loc, valueToTile, r); in makeTiledShape()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 135 sizes.push_back(linalg::createOrFoldDimOp(builder, loc, src, i)); in sizesFromSrc()
832 hi.push_back(linalg::createOrFoldDimOp(rewriter, loc, values, 0)); in matchAndRewrite()
837 hi.push_back(linalg::createOrFoldDimOp(rewriter, loc, src, i)); in matchAndRewrite()
|
| H A D | Sparsification.cpp | 530 Value up = linalg::createOrFoldDimOp(builder, loc, t->get(), p); in genBuffers()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 124 Value dim = createOrFoldDimOp(b, loc, shapeDim.shape, shapeDim.dimension); in fuse()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToSCF/ |
| H A D | VectorToSCF.cpp | 171 vector::createOrFoldDimOp(b, loc, xferOp.getSource(), *dim); in generateInBoundsCheck()
|