diff options
| author | Andrzej Warzyński <andrzej.warzynski@arm.com> | 2024-10-15 19:24:43 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-15 19:24:43 +0100 |
| commit | a758bcdbd92efb64a3482eb95d2769d74e33f5bb (patch) | |
| tree | 4ebf3edf1e40dfcb62648d32f62bf378e6ab8aed /mlir/python | |
| parent | 87db0c06013412dd34953b0aaa3c3c02e45bd571 (diff) | |
[mlir][td] Rename pack_paddings in structured.pad (#111036)
The pack_paddings attribute in the structure.pad TD Op is used to set
the `nofold` attribute in the generated tensor.pad Op. The current name
is confusing and suggests that there's a relation with the tensor.pack
Op. This patch renames it as `nofold_flags` to better match the actual
usage.
Diffstat (limited to 'mlir/python')
| -rw-r--r-- | mlir/python/mlir/dialects/transform/structured.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/python/mlir/dialects/transform/structured.py b/mlir/python/mlir/dialects/transform/structured.py index 41051c0d5b2f..f6111f516f8c 100644 --- a/mlir/python/mlir/dialects/transform/structured.py +++ b/mlir/python/mlir/dialects/transform/structured.py @@ -377,7 +377,7 @@ class PadOp(PadOp): pad_to_multiple_of: Optional[Union[DynamicIndexList, ArrayAttr]] = None, padding_values: Optional[Union[ArrayAttr, Sequence[Attribute]]] = None, padding_dimensions: OptionalIntList = None, - pack_paddings: OptionalIntList = None, + nofold_flags: OptionalIntList = None, transpose_paddings: Optional[ Union[ArrayAttr, Sequence[Union[ArrayAttr, IntOrAttrList]]] ] = None, @@ -407,7 +407,7 @@ class PadOp(PadOp): padding_values=padding_values, padding_dimensions=padding_dimensions, static_pad_to_multiple_of=static_pad_to_multiple_of, - pack_paddings=pack_paddings, + nofold_flags=nofold_flags, transpose_paddings=transpose_paddings, copy_back_op=copy_back_op, loc=loc, |