mlir.dialects.transform.structured

Classes

BufferizeToAllocationOp	Specialization for BufferizeToAllocationOp class.
DecomposeOp	Specialization for DecomposeOp class.
FuseIntoContainingOp	Specialization for FuseIntoContainingOp class.
FuseOp	Specialization for FuseOp class.
GeneralizeOp	Specialization for GeneralizeOp class.
InterchangeOp	Specialization for InterchangeOp class.
MapCopyToThreadsOp	Specialization for MapCopyToThreadsOp class.
VectorizeOp	Specialization for VectorizeOp class.
MatchOp	Specialization for MatchOp class.
MultiTileSizesOp	Specialization for MultiTileSizesOp class.
PadOp	Specialization for PadOp class.
ScalarizeOp	Specialization for ScalarizeOp class.
SplitOp	Specialization for SplitOp class.
TileUsingForOp	Specialization for TileUsingForOp class.
TileUsingForallOp	Specialization for TileUsingForallOp class.
VectorizeChildrenAndApplyPatternsOp	Specialization for VectorizeChildrenAndApplyPatternsOp class.

Module Contents

class mlir.dialects.transform.structured.BufferizeToAllocationOp(target: Operation | OpView | Value, *, memory_space: int | str | Attribute | None = None, memcpy_op: str | None = None, alloc_op: str | None = None, bufferize_destination_only: bool | None = None, loc=None, ip=None)

Bases: BufferizeToAllocationOp

Specialization for BufferizeToAllocationOp class.

class mlir.dialects.transform.structured.DecomposeOp(target: Operation | Value, *, loc=None, ip=None)

Bases: DecomposeOp

Specialization for DecomposeOp class.

class mlir.dialects.transform.structured.FuseIntoContainingOp(fused_op_type: Type, new_containing_op_type: Type, producer_op: Operation | OpView | Value, containing_op: Operation | OpView | Value, *, loc=None, ip=None)

class mlir.dialects.transform.structured.FuseIntoContainingOp(producer_op: Operation | OpView | Value, containing_op: Operation | OpView | Value, *, loc=None, ip=None)

Bases: FuseIntoContainingOp

Specialization for FuseIntoContainingOp class.

class mlir.dialects.transform.structured.FuseOp(loop_types: Type | Sequence[Type], target: Operation | Value | OpView, *, tile_sizes: mlir.dialects._ods_common.MixedValues | None = None, tile_interchange: mlir.dialects._ods_common.MixedValues | None = None, apply_cleanup: bool = False, use_forall: bool = False, loc=None, ip=None)

class mlir.dialects.transform.structured.FuseOp(target: Operation | Value | OpView, *, tile_sizes: mlir.dialects._ods_common.MixedValues | None = None, tile_interchange: mlir.dialects._ods_common.MixedValues | None = None, apply_cleanup: bool = False, use_forall: bool = False, loc=None, ip=None)

Bases: FuseOp

Specialization for FuseOp class.

class mlir.dialects.transform.structured.GeneralizeOp(target: Operation | Value, *, loc=None, ip=None)

Bases: GeneralizeOp

Specialization for GeneralizeOp class.

class mlir.dialects.transform.structured.InterchangeOp(target: Operation | Value, *, iterator_interchange: mlir.dialects._ods_common.OptionalIntList = None, loc=None, ip=None)

Bases: InterchangeOp

Specialization for InterchangeOp class.

class mlir.dialects.transform.structured.MapCopyToThreadsOp(forall_op_type: Type, tiled_op_type: Type, target: Operation | OpView | Value, *, total_num_threads: int | IntegerAttr, desired_bit_alignment: int | IntegerAttr, loc=None, ip=None)

class mlir.dialects.transform.structured.MapCopyToThreadsOp(target: Operation | OpView | Value, *, total_num_threads: int | IntegerAttr, desired_bit_alignment: int | IntegerAttr, loc=None, ip=None)

Bases: MapCopyToThreadsOp

Specialization for MapCopyToThreadsOp class.

class mlir.dialects.transform.structured.VectorizeOp(target: Operation | OpView | Value, vector_sizes: mlir.dialects._ods_common.DynamicIndexList | ArrayAttr | None = None, *, vectorize_nd_extract: bool | None = None, scalable_sizes: mlir.dialects._ods_common.OptionalBoolList = None, static_vector_sizes: mlir.dialects._ods_common.OptionalIntList = None, loc=None, ip=None)

Bases: VectorizeOp

Specialization for VectorizeOp class.

class mlir.dialects.transform.structured.MatchOp(results_, target, *, ops=None, interface=None, op_attrs=None, filter_result_type=None, filter_operand_types=None, loc=None, ip=None)

Bases: MatchOp

Specialization for MatchOp class.

classmethod match_op_names(target: Operation | Value, names: str | Sequence[str], *, loc=None, ip=None)

classmethod match_op_names(result_type: Type, target: Operation | Value, names: str | Sequence[str], *, loc=None, ip=None)

class mlir.dialects.transform.structured.MultiTileSizesOp(result_type: Type, target: Operation | Value, *, dimension: int | IntegerAttr, target_size: int | IntegerAttr, divisor: int | IntegerAttr | None | None = None, loc=None, ip=None)

Bases: MultiTileSizesOp

Specialization for MultiTileSizesOp class.

class mlir.dialects.transform.structured.PadOp(target: Operation | OpView | Value, *, pad_to_multiple_of: mlir.dialects._ods_common.DynamicIndexList | ArrayAttr | None = None, padding_values: ArrayAttr | Sequence[Attribute] | None = None, padding_dimensions: mlir.dialects._ods_common.OptionalIntList = None, nofold_flags: mlir.dialects._ods_common.OptionalIntList = None, transpose_paddings: ArrayAttr | Sequence[ArrayAttr | mlir.dialects._ods_common.IntOrAttrList] | None = None, copy_back_op: str | StringAttr | None = None, loc=None, ip=None)

Bases: PadOp

Specialization for PadOp class.

class mlir.dialects.transform.structured.ScalarizeOp(target: Operation | Value, *, loc=None, ip=None)

Bases: ScalarizeOp

Specialization for ScalarizeOp class.

class mlir.dialects.transform.structured.SplitOp(target: Operation | Value, dimension: int | Attribute, chunk_sizes: int | Operation | Value | Attribute, *, loc=None, ip=None)

Bases: SplitOp

Specialization for SplitOp class.

class mlir.dialects.transform.structured.TileUsingForOp(loop_types: Type | List[Type], target: Operation | Value, *, sizes: mlir.dialects._ods_common.DynamicIndexList | ArrayAttr | None = None, interchange: mlir.dialects._ods_common.OptionalIntList = None, loc=None, ip=None)

class mlir.dialects.transform.structured.TileUsingForOp(target: Operation | Value | OpView, *, sizes: mlir.dialects._ods_common.DynamicIndexList | ArrayAttr | None = None, interchange: mlir.dialects._ods_common.OptionalIntList = None, loc=None, ip=None)

Bases: TileUsingForOp

Specialization for TileUsingForOp class.

class mlir.dialects.transform.structured.TileUsingForallOp(loops_type: Type, tiled_op_type: Type, target: Operation | Value | OpView, *, num_threads: mlir.dialects._ods_common.MixedValues | None = None, tile_sizes: mlir.dialects._ods_common.MixedValues = None, mapping=None, loc=None, ip=None)

class mlir.dialects.transform.structured.TileUsingForallOp(target: Operation | Value | OpView, *, num_threads: mlir.dialects._ods_common.MixedValues | None = None, tile_sizes: mlir.dialects._ods_common.MixedValues = None, mapping=None, loc=None, ip=None)

Bases: TileUsingForallOp

Specialization for TileUsingForallOp class.

class mlir.dialects.transform.structured.VectorizeChildrenAndApplyPatternsOp(target: Operation | Value, *, disable_multi_reduction_to_contract_patterns: bool = False, disable_transfer_permutation_map_lowering_patterns: bool = False, vectorize_nd_extract: bool = False, vectorize_padding: bool = False, flatten_1d_depthwise_conv: bool = False, fold_type_extensions_into_contract: bool = False, loc=None, ip=None)

Bases: VectorizeChildrenAndApplyPatternsOp

Specialization for VectorizeChildrenAndApplyPatternsOp class.