Home
last modified time | relevance | path

Searched refs:static_sizes (Results 1 – 10 of 10) sorted by relevance

/llvm-project-15.0.7/mlir/python/mlir/dialects/
H A D_structured_transform_ops_ext.py228 static_sizes = []
235 static_sizes.append(IntegerAttr.get(i64_type, size))
237 static_sizes.append(size)
239 static_sizes.append(
242 sizes_attr = ArrayAttr.get(static_sizes)
250 static_sizes=sizes_attr,
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DViewLikeInterface.td41 3. `static_offsets`, resp. `static_sizes` and `static_strides` integer
47 1. `static_offsets`, `static_sizes` and `static_strides` have length
51 3. if an entry of `static_offsets` (resp. `static_sizes`,
80 Return the expected rank of each of the`static_offsets`, `static_sizes`
140 /*methodName=*/"static_sizes",
144 return $_op.static_sizes();
181 return ::mlir::getMixedSizes($_op, $_op.static_sizes(), $_op.sizes());
221 ::llvm::APInt v = *(static_sizes()
266 ::llvm::APInt v = *(static_sizes().
317 static_sizes().template cast<::mlir::ArrayAttr>(), ::mlir::ShapedType::isDynamic, idx);
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgOps.td43 (ins Variadic<Index>:$sizes, I64ArrayAttr:$static_sizes);
48 custom<OperandsOrIntegersSizesList>($sizes, $static_sizes) attr-dict
54 return "static_sizes";
67 APInt v = *(static_sizes().getAsValueRange<IntegerAttr>().begin() + idx);
75 APInt v = *(static_sizes().
86 static_sizes().getValue().begin(),
87 static_sizes().getValue().begin() + idx,
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DViewLikeInterface.cpp63 op.static_sizes(), op.sizes(), in verifyOffsetSizeAndStrideOp()
166 if (a.static_sizes().size() != b.static_sizes().size()) in sameOffsetsSizesAndStrides()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensorOps.td200 `static_offsets`, `static_sizes` and `static_strides` arguments. A special
250 I64ArrayAttr:$static_sizes,
258 custom<OperandsOrIntegersSizesList>($sizes, $static_sizes)
340 /// Return the expected rank of each of the`static_offsets`, `static_sizes`
536 `static_offsets`, `static_sizes` and `static_strides` arguments. A special
576 I64ArrayAttr:$static_sizes,
584 custom<OperandsOrIntegersSizesList>($sizes, $static_sizes)
617 /// Return the expected rank of each of the`static_offsets`, `static_sizes`
1112 `static_offsets`, `static_sizes` and `static_strides` arguments. A special
1142 I64ArrayAttr:$static_sizes,
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/
H A Dops.py28 def static_sizes(): function
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1204 static_sizes().cast<ArrayAttr>(), in verify()
1208 *this, "sizes", resultType.getRank(), static_sizes(), sizes(), in verify()
1212 if (static_sizes().size() != static_cast<unsigned>(resultType.getRank())) in verify()
1234 for (Attribute attr : static_sizes()) { in getMixedSizes()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/
H A DMemRefOps.td1020 I64ArrayAttr:$static_sizes,
1028 custom<OperandsOrIntegersSizesList>($sizes, $static_sizes) `` `,` `strides`
1064 /// Return the expected rank of each of the`static_offsets`, `static_sizes`
1461 `static_offsets`, `static_sizes` and `static_strides` arguments. A special
1583 I64ArrayAttr:$static_sizes,
1590 custom<OperandsOrIntegersSizesList>($sizes, $static_sizes)
1667 /// Return the expected rank of each of the`static_offsets`, `static_sizes`
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.td617 `static_sizes` attribute must be set to `ShapedType::kDynamicSize`. Only
620 `static_sizes` attribute. A static size of `0` indicates that the dimension
647 DefaultValuedAttr<I64ArrayAttr, "{}">:$static_sizes,
/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dinvalid.mlir362 %1 = "linalg.init_tensor"(%arg0) {static_sizes = [4, -1, -1, 5]} : (index) -> tensor<4x?x?x5xf32>