| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | TypeUtilities.cpp | 63 int64_t dim2 = std::get<1>(dims); in verifyCompatibleShape() local
64 if (!ShapedType::isDynamic(dim1) && !ShapedType::isDynamic(dim2) && in verifyCompatibleShape()
65 dim1 != dim2) in verifyCompatibleShape()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | Traits.cpp | 197 auto isCompatible = [](int64_t dim1, int64_t dim2) { in isCompatibleInferredReturnShape() argument
202 return dim1 == dim2 || dim1 == -1 || dim2 == -1 || dim1 == 1; in isCompatibleInferredReturnShape()
|
| /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):
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 363 AffineExpr dim1, dim2; in makeComposedExtractSliceOp() local
364 bindDims(b.getContext(), dim1, dim2); in makeComposedExtractSliceOp()
370 makeComposedAffineApply(b, loc, dim1 + dim2, offsetValues).getResult(); in makeComposedExtractSliceOp()
865 AffineExpr dim0, dim1, dim2; in makeTiledShape() local
866 bindDims(builder.getContext(), dim0, dim1, dim2); in makeTiledShape()
889 {ArrayRef<AffineExpr>{dim1 - dim2, dim0}}) in makeTiledShape()
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | loops.mlir | 748 // CHECK: %[[dim2:.*]] = memref.dim %[[arg2]], %[[c0]] : memref<?x?xf32>
750 // CHECK: scf.for %[[arg3:.*]] = %[[c0]] to %[[dim2]] step %[[c1]] {
773 // CHECKPARALLEL: %[[dim2:.*]] = memref.dim %[[arg2]], %[[c0]] : memref<?x?xf32>
775 // CHECKPARALLEL: scf.parallel (%[[arg3:.*]], %[[arg4:.*]]) = (%[[c0]], %[[c0]]) to (%[[dim2]…
803 // CHECK: %[[dim2:.*]] = memref.dim %[[arg1]], %[[c2]] : memref<?x?x?xf32>
812 // CHECK: scf.for %[[arg8:.*]] = %[[c0]] to %[[dim2]] step %[[c1]] {
834 // CHECKPARALLEL: %[[dim2:.*]] = memref.dim %[[arg1]], %[[c2]] : memref<?x?x?xf32>
841 // CHECKPARALLEL: scf.for %[[arg8:.*]] = %[[c0]] to %[[dim2]] step %[[c1]] {
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 892 auto dim2 = shape.getDimSize(i); in REDUCE_SHAPE_INFER() local
896 resolvedDim = dim2; in REDUCE_SHAPE_INFER()
897 } else if (dim2 == 1) { in REDUCE_SHAPE_INFER()
899 } else if (dim1 != dim2) { in REDUCE_SHAPE_INFER()
|
| /llvm-project-15.0.7/polly/lib/External/isl/ |
| H A D | isl_space.c | 3304 isl_size dim1, dim2; in isl_space_cmp_type() local
3308 dim2 = isl_space_dim(space2, type); in isl_space_cmp_type()
3309 if (dim1 < 0 || dim2 < 0) in isl_space_cmp_type()
3311 if (dim1 != dim2) in isl_space_cmp_type()
3312 return dim1 - dim2; in isl_space_cmp_type()
|
| H A D | isl_union_map.c | 4109 isl_size dim1, dim2; in check_union_map_space_equal_dim() local
4112 dim2 = isl_space_dim(space, isl_dim_param); in check_union_map_space_equal_dim()
4113 if (dim1 < 0 || dim2 < 0) in check_union_map_space_equal_dim()
4115 if (dim1 == dim2) in check_union_map_space_equal_dim()
|
| H A D | isl_map.c | 3716 isl_size dim1, dim2, nparam1, nparam2; in isl_basic_map_intersect() local
3728 if (dim1 == nparam1 && dim2 != nparam2) in isl_basic_map_intersect()
3731 if (dim2 != nparam2 && in isl_basic_map_intersect()
3883 isl_size dim2, nparam2; in map_intersect_internal() local
3916 dim2 = isl_map_dim(map2, isl_dim_all); in map_intersect_internal()
3918 if (dim2 < 0 || nparam2 < 0) in map_intersect_internal()
3920 if (dim2 != nparam2) in map_intersect_internal()
5868 isl_size dim1, dim2; in check_map_space_equal_total_dim() local
5871 dim2 = isl_space_dim(space, isl_dim_all); in check_map_space_equal_total_dim()
5872 if (dim1 < 0 || dim2 < 0) in check_map_space_equal_total_dim()
[all …]
|
| /llvm-project-15.0.7/flang/test/Fir/ |
| H A D | convert-to-llvm.fir | 1811 // Calculate nelems in dim2
|