| /llvm-project-15.0.7/mlir/unittests/Dialect/MemRef/ |
| H A D | InferShapeTest.cpp | 25 AffineExpr dim0; in TEST() local
26 bindDims(&ctx, dim0); in TEST()
28 MemRefType::get({2}, b.getIndexType(), AffineMap::get(1, 0, dim0 + 13)); in TEST()
36 AffineExpr dim0, dim1; in TEST() local
37 bindDims(&ctx, dim0, dim1); in TEST()
39 AffineMap::get(2, 0, 1000 * dim0 + dim1)); in TEST()
43 MemRefType::get({2}, b.getIndexType(), AffineMap::get(1, 0, dim0 + 2003)); in TEST()
50 AffineExpr dim0, dim1; in TEST() local
51 bindDims(&ctx, dim0, dim1); in TEST()
53 AffineMap::get(2, 0, 1000 * dim0 + dim1)); in TEST()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 88 AffineExpr dim0, dim1; in bubbleUpPadSlice() local
89 bindDims(b.getContext(), dim0, dim1); in bubbleUpPadSlice()
91 auto addMap = AffineMap::get(2, 0, {dim0 + dim1}); in bubbleUpPadSlice()
96 auto subMap = AffineMap::get(2, 0, {dim0 - dim1}); in bubbleUpPadSlice()
|
| /llvm-project-15.0.7/mlir/test/Dialect/Tensor/ |
| H A D | split-padding.mlir | 8 ^bb0(%dim0: index, %dim1: index, %dim2: index):
24 ^bb0(%dim0: index, %dim1: index, %dim2: index):
|
| H A D | bufferize.mlir | 300 // CHECK-DAG: %[[dim0:.*]] = memref.dim %[[m1]], %[[c0]]
302 // CHECK: %[[alloc:.*]] = memref.alloc(%[[dim0]], %[[dim1]])
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | loops.mlir | 704 // CHECK: %[[dim0:.*]] = memref.dim %[[arg1]], %[[c0]] : memref<?xf32>
707 // CHECK: scf.for %[[m:.*]] = %[[c0]] to %[[dim0]] step %[[c1]] {
722 // CHECKPARALLEL: %[[dim0:.*]] = memref.dim %[[arg1]], %[[c0]] : memref<?xf32>
725 // CHECKPARALLEL: scf.for %[[m:.*]] = %[[c0]] to %[[dim0]] step %[[c1]] {
746 // CHECK: %[[dim0:.*]] = memref.dim %[[arg1]], %[[c0]] : memref<?x?xf32>
752 // CHECK: scf.for %[[arg5:.*]] = %[[c0]] to %[[dim0]] step %[[c1]] {
771 // CHECKPARALLEL: %[[dim0:.*]] = memref.dim %[[arg1]], %[[c0]] : memref<?x?xf32>
776 // CHECKPARALLEL: scf.for %[[arg5:.*]] = %[[c0]] to %[[dim0]] step %[[c1]] {
801 // CHECK: %[[dim0:.*]] = memref.dim %[[arg1]], %[[c0]] : memref<?x?x?xf32>
810 // CHECK: scf.for %[[arg6:.*]] = %[[c0]] to %[[dim0]] step %[[c1]] {
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/X86Vector/Transforms/ |
| H A D | AVXTranspose.cpp | 208 static bool areDimsTransposedIn2DSlice(int64_t dim0, int64_t dim1, in areDimsTransposedIn2DSlice() argument
214 if (permDim == dim0) in areDimsTransposedIn2DSlice()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 865 AffineExpr dim0, dim1, dim2; in makeTiledShape() local
866 bindDims(builder.getContext(), dim0, dim1, dim2); in makeTiledShape()
875 AffineMap::inferFromExprList({ArrayRef<AffineExpr>{dim0 - 1}}) in makeTiledShape()
878 AffineMap::inferFromExprList({ArrayRef<AffineExpr>{dim0 + 1}}) in makeTiledShape()
889 {ArrayRef<AffineExpr>{dim1 - dim2, dim0}}) in makeTiledShape()
|
| /llvm-project-15.0.7/mlir/test/Dialect/Affine/ |
| H A D | loop-tiling.mlir | 161 %dim0 = memref.dim %arg0, %c0 : memref<?xf32>
162 affine.for %i0 = 0 to affine_map<()[s0, s1] -> (s0 + s1)> ()[%dim0, %limit] {
|
| H A D | loop-tiling-parametric.mlir | 220 %dim0 = memref.dim %arg0, %c0 : memref<?xf32>
221 affine.for %i0 = 0 to affine_map<()[s0, s1] -> (s0 + s1)> ()[%dim0, %limit] {
|