| # Originally imported via: |
| # stubgen {...} -m mlir._mlir_libs._mlir.ir |
| # Local modifications: |
| # * Rewrite references to 'mlir.ir.' to local types |
| # * Add __all__ with the following incantation: |
| # egrep '^class ' ir.pyi | awk -F ' |:|\\(' '{print " \"" $2 "\","}' |
| # * Local edits to signatures and types that MyPy did not auto detect (or |
| # detected incorrectly). |
| |
| from typing import Any, Callable, ClassVar, Dict, List, Optional, Sequence |
| |
| from typing import overload |
| |
| __all__ = [ |
| "AffineAddExpr", |
| "AffineBinaryExpr", |
| "AffineCeilDivExpr", |
| "AffineConstantExpr", |
| "AffineDimExpr", |
| "AffineExpr", |
| "AffineExprList", |
| "AffineFloorDivExpr", |
| "AffineMap", |
| "AffineMapAttr", |
| "AffineModExpr", |
| "AffineMulExpr", |
| "AffineSymbolExpr", |
| "ArrayAttr", |
| "ArrayAttributeIterator", |
| "Attribute", |
| "BF16Type", |
| "Block", |
| "BlockArgument", |
| "BlockArgumentList", |
| "BlockIterator", |
| "BlockList", |
| "BoolAttr", |
| "ComplexType", |
| "Context", |
| "DenseElementsAttr", |
| "DenseFPElementsAttr", |
| "DenseIntElementsAttr", |
| "Dialect", |
| "DialectDescriptor", |
| "Dialects", |
| "DictAttr", |
| "F16Type", |
| "F32Type", |
| "F64Type", |
| "FlatSymbolRefAttr", |
| "FloatAttr", |
| "FunctionType", |
| "IndexType", |
| "InferTypeOpInterface", |
| "InsertionPoint", |
| "IntegerAttr", |
| "IntegerSet", |
| "IntegerSetConstraint", |
| "IntegerSetConstraintList", |
| "IntegerType", |
| "Location", |
| "MemRefType", |
| "Module", |
| "NamedAttribute", |
| "NoneType", |
| "OpAttributeMap", |
| "OpOperandList", |
| "OpResult", |
| "OpResultList", |
| "OpView", |
| "Operation", |
| "OperationIterator", |
| "OperationList", |
| "RankedTensorType", |
| "Region", |
| "RegionIterator", |
| "RegionSequence", |
| "ShapedType", |
| "StringAttr", |
| "SymbolTable", |
| "TupleType", |
| "Type", |
| "TypeAttr", |
| "UnitAttr", |
| "UnrankedMemRefType", |
| "UnrankedTensorType", |
| "Value", |
| "VectorType", |
| "_GlobalDebug", |
| "_OperationBase", |
| ] |
| |
| # Base classes: declared first to simplify declarations below. |
| class _OperationBase: |
| def detach_from_parent(self) -> "OpView": ... |
| def get_asm(self, binary: bool = False, large_elements_limit: Optional[int] = None, enable_debug_info: bool = False, pretty_debug_info: bool = False, print_generic_op_form: bool = False, use_local_scope: bool = False, assume_verified: bool = False) -> object: ... |
| def move_after(self, other: "_OperationBase") -> None: ... |
| def move_before(self, other: "_OperationBase") -> None: ... |
| def print(self, file: Optional[Any] = None, binary: bool = False, large_elements_limit: Optional[int] = None, enable_debug_info: bool = False, pretty_debug_info: bool = False, print_generic_op_form: bool = False, use_local_scope: bool = False, assume_verified: bool = False) -> None: ... |
| def verify(self) -> bool: ... |
| @overload |
| def __eq__(self, arg0: "_OperationBase") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def attributes(self) -> "OpAttributeMap": ... |
| @property |
| def location(self) -> "Location": ... |
| @property |
| def operands(self) -> "OpOperandList": ... |
| @property |
| def regions(self) -> "RegionSequence": ... |
| @property |
| def result(self) -> "OpResult": ... |
| @property |
| def results(self) -> "OpResultList": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineExpr: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def _CAPICreate(self) -> AffineExpr: ... |
| def compose(self, arg0) -> AffineExpr: ... |
| def dump(self) -> None: ... |
| def get_add(self, *args, **kwargs) -> Any: ... |
| def get_ceil_div(self, *args, **kwargs) -> Any: ... |
| def get_constant(self, *args, **kwargs) -> Any: ... |
| def get_dim(self, *args, **kwargs) -> Any: ... |
| def get_floor_div(self, *args, **kwargs) -> Any: ... |
| def get_mod(self, *args, **kwargs) -> Any: ... |
| def get_mul(self, *args, **kwargs) -> Any: ... |
| def get_symbol(self, *args, **kwargs) -> Any: ... |
| def __add__(self, other) -> Any: ... |
| @overload |
| def __eq__(self, arg0: AffineExpr) -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| def __mod__(self, other) -> Any: ... |
| def __mul__(self, other) -> Any: ... |
| def __radd__(self, other) -> Any: ... |
| def __rmod__(self, other) -> Any: ... |
| def __rmul__(self, other) -> Any: ... |
| def __rsub__(self, other) -> Any: ... |
| def __sub__(self, other) -> Any: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> object: ... |
| |
| class Attribute: |
| def __init__(self, cast_from_type: "Attribute") -> None: ... |
| def _CAPICreate(self) -> "Attribute": ... |
| def dump(self) -> None: ... |
| def get_named(self, *args, **kwargs) -> Any: ... |
| @staticmethod |
| def parse(asm: str, context: Optional["Context"] = None) -> Any: ... |
| @overload |
| def __eq__(self, arg0: "Attribute") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> "Context"": ... |
| @property |
| def type(self) -> "Type": ... |
| |
| class Type: |
| def __init__(self, cast_from_type: "Type") -> None: ... |
| def _CAPICreate(self) -> Type: ... |
| def dump(self) -> None: ... |
| @staticmethod |
| def parse(asm: str, context: Optional["Context"] = None) -> "Type": ... |
| @overload |
| def __eq__(self, arg0: "Type") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> "Context": ... |
| |
| class Value: |
| def _CAPICreate(self) -> "Value": ... |
| def dump(self) -> None: ... |
| @overload |
| def __eq__(self, arg0: "Value") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> "Context": ... |
| @property |
| def owner(self) -> "_OperationBase": ... |
| @property |
| def type(self) -> "Type": ... |
| |
| |
| # Classes with no particular order sensitivity in alpha order. |
| # TODO: Auto-generated. Audit and fix. |
| class AffineAddExpr(AffineBinaryExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineBinaryExpr(AffineExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def lhs(self) -> AffineExpr: ... |
| @property |
| def rhs(self) -> AffineExpr: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineCeilDivExpr(AffineBinaryExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineConstantExpr(AffineExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def value(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineDimExpr(AffineExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def position(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineExprList: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def __add__(self, arg0: AffineExprList) -> List[AffineExpr]: ... |
| @overload |
| def __getitem__(self, arg0: int) -> AffineExpr: ... |
| @overload |
| def __getitem__(self, arg0: slice) -> AffineExprList: ... |
| def __len__(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineFloorDivExpr(AffineBinaryExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineMap: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def _CAPICreate(self) -> AffineMap: ... |
| def compress_unused_symbols(self, *args, **kwargs) -> Any: ... |
| def dump(self) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def get_constant(self, *args, **kwargs) -> Any: ... |
| def get_empty(self, *args, **kwargs) -> Any: ... |
| def get_identity(self, *args, **kwargs) -> Any: ... |
| def get_major_submap(self, n_results: int) -> AffineMap: ... |
| def get_minor_identity(self, *args, **kwargs) -> Any: ... |
| def get_minor_submap(self, n_results: int) -> AffineMap: ... |
| def get_permutation(self, *args, **kwargs) -> Any: ... |
| def get_submap(self, result_positions: List[int]) -> AffineMap: ... |
| def replace(self, expr: AffineExpr, replacement: AffineExpr, n_result_dims: int, n_result_syms: int) -> AffineMap: ... |
| @overload |
| def __eq__(self, arg0: AffineMap) -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> object: ... |
| @property |
| def is_permutation(self) -> bool: ... |
| @property |
| def is_projected_permutation(self) -> bool: ... |
| @property |
| def n_dims(self) -> int: ... |
| @property |
| def n_inputs(self) -> int: ... |
| @property |
| def n_symbols(self) -> int: ... |
| @property |
| def results(self) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineMapAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineModExpr(AffineBinaryExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineMulExpr(AffineBinaryExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class AffineSymbolExpr(AffineExpr): |
| def __init__(self, expr: AffineExpr) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def position(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class ArrayAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def __add__(self, arg0: list) -> ArrayAttr: ... |
| def __getitem__(self, arg0: int) -> Attribute: ... |
| def __iter__(self) -> Any: ... |
| def __len__(self) -> int: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class ArrayAttributeIterator: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def __iter__(self) -> ArrayAttributeIterator: ... |
| def __next__(self) -> Attribute: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class BF16Type(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| class Block: |
| __hash__: ClassVar[None] = ... |
| def append(self, operation: _OperationBase) -> None: ... |
| def create_after(self, *args: "Type") -> "Block": ... |
| @staticmethod |
| def create_at_start(parent: "Region", arg_types: List["Type"]) -> "Block": ... |
| def create_before(self, *args: "Type") -> "Block": ... |
| @overload |
| def __eq__(self, arg0: "Block") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __iter__(self) -> Any: ... |
| @property |
| def arguments(self) -> "BlockArgumentList": ... |
| @property |
| def operations(self) -> "OperationList": ... |
| @property |
| def owner(self) -> "OpView": ... |
| @property |
| def region(self) -> "Region": ... |
| |
| class BlockArgument(Value): |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def set_type(self, type: "Type") -> None: ... |
| @property |
| def arg_number(self) -> int: ... |
| @property |
| def owner(self) -> "Block": ... |
| |
| class BlockArgumentList: |
| def __add__(self, arg0: "BlockArgumentList") -> List["BlockArgument"]: ... |
| @overload |
| def __getitem__(self, arg0: int) -> "BlockArgument": ... |
| @overload |
| def __getitem__(self, arg0: slice) -> "BlockArgumentList": ... |
| def __len__(self) -> int: ... |
| @property |
| def types(self) -> List["Type"]: ... |
| |
| class BlockIterator: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def __iter__(self) -> "BlockIterator": ... |
| def __next__(self) -> "Block": ... |
| |
| class BlockList: |
| def append(self, *args) -> "Block": ... |
| def __getitem__(self, arg0: int) -> "Block": ... |
| def __iter__(self) -> "BlockIterator": ... |
| def __len__(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class BoolAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> bool: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class ComplexType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def element_type(self) -> Type: ... |
| |
| class Context: |
| current: ClassVar["Context"] = ... # read-only |
| allow_unregistered_dialects: bool |
| def __init__(self) -> None: ... |
| def _CAPICreate(self) -> object: ... |
| def _get_context_again(self) -> "Context": ... |
| @staticmethod |
| def _get_live_count() -> int: ... |
| def _get_live_module_count(self) -> int: ... |
| def _get_live_operation_count(self) -> int: ... |
| def enable_multithreading(self, enable: bool) -> None: ... |
| def get_dialect_descriptor(name: dialect_name: str) -> "DialectDescriptor": ... |
| def is_registered_operation(self, operation_name: str) -> bool: ... |
| def __enter__(self) -> "Context": ... |
| def __exit__(self, arg0: object, arg1: object, arg2: object) -> None: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def d(self) -> "Dialects": ... |
| @property |
| def dialects(self) -> "Dialects": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class DenseElementsAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def get_splat(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def __len__(self) -> int: ... |
| @property |
| def is_splat(self) -> bool: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class DenseFPElementsAttr(DenseElementsAttr): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def __getitem__(self, arg0: int) -> float: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class DenseIntElementsAttr(DenseElementsAttr): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def __getitem__(self, arg0: int) -> int: ... |
| @property |
| def type(self) -> Type: ... |
| |
| class Dialect: |
| def __init__(self, descriptor: "DialectDescriptor") -> None: ... |
| @property |
| def descriptor(self) -> "DialectDescriptor": ... |
| |
| class DialectDescriptor: |
| @property |
| def namespace(self) -> str: ... |
| |
| class Dialects: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def __getattr__(self, arg0: str) -> "Dialect": ... |
| def __getitem__(self, arg0: str) -> "Dialect": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class DictAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| def __contains__(self, arg0: str) -> bool: ... |
| @overload |
| def __getitem__(self, arg0: str) -> Attribute: ... |
| @overload |
| def __getitem__(self, arg0: int) -> NamedAttribute: ... |
| def __len__(self) -> int: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class F16Type(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class F32Type(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class F64Type(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class FlatSymbolRefAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> str: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class FloatAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def get_f32(self, *args, **kwargs) -> Any: ... |
| def get_f64(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> float: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class FunctionType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def inputs(self) -> list: ... |
| @property |
| def results(self) -> list: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IndexType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| class InferTypeOpInterface: |
| def __init__(self, object: object, context: Optional["Context"] = None) -> None: ... |
| def inferReturnTypes(self, operands: Optional[List["Value"]] = None, attributes: Optional["Attribute"] = None, regions: Optional[List["Region"]] = None, context: Optional["Context"] = None, loc: Optional["Location"] = None) -> List[Type]: ... |
| @property |
| def operation(self) -> "_OperationBase": ... |
| @property |
| def opview(self) -> "OpView": ... |
| |
| class InsertionPoint: |
| current: ClassVar["InsertionPoint"] = ... # read-only |
| @overload |
| def __init__(self, block: "Block") -> None: ... |
| @overload |
| def __init__(self, beforeOperation: "_OperationBase") -> None: ... |
| @staticmethod |
| def at_block_begin(block: "Block") -> "InsertionPoint": ... |
| @staticmethod |
| def at_block_terminator(block: "Block") -> "InsertionPoint": ... |
| def insert(self, operation: "_OperationBase") -> None: ... |
| def __enter__(self) -> "InsertionPoint": ... |
| def __exit__(self, arg0: object, arg1: object, arg2: object) -> None: ... |
| @property |
| def block(self) -> "Block": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IntegerAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IntegerSet: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def _CAPICreate(self) -> IntegerSet: ... |
| def dump(self) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def get_empty(self, *args, **kwargs) -> Any: ... |
| def get_replaced(self, dim_exprs: list, symbol_exprs: list, num_result_dims: int, num_result_symbols: int) -> IntegerSet: ... |
| @overload |
| def __eq__(self, arg0: IntegerSet) -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __hash__(self) -> int: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def constraints(self) -> Any: ... |
| @property |
| def context(self) -> object: ... |
| @property |
| def is_canonical_empty(self) -> bool: ... |
| @property |
| def n_dims(self) -> int: ... |
| @property |
| def n_equalities(self) -> int: ... |
| @property |
| def n_inequalities(self) -> int: ... |
| @property |
| def n_inputs(self) -> int: ... |
| @property |
| def n_symbols(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IntegerSetConstraint: |
| def __init__(self, *args, **kwargs) -> None: ... |
| @property |
| def expr(self) -> AffineExpr: ... |
| @property |
| def is_eq(self) -> bool: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IntegerSetConstraintList: |
| def __init__(self, *args, **kwargs) -> None: ... |
| def __add__(self, arg0: IntegerSetConstraintList) -> List[IntegerSetConstraint]: ... |
| @overload |
| def __getitem__(self, arg0: int) -> IntegerSetConstraint: ... |
| @overload |
| def __getitem__(self, arg0: slice) -> IntegerSetConstraintList: ... |
| def __len__(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class IntegerType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get_signed(self, *args, **kwargs) -> Any: ... |
| def get_signless(self, *args, **kwargs) -> Any: ... |
| def get_unsigned(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def is_signed(self) -> bool: ... |
| @property |
| def is_signless(self) -> bool: ... |
| @property |
| def is_unsigned(self) -> bool: ... |
| @property |
| def width(self) -> int: ... |
| |
| class Location: |
| current: ClassVar["Location"] = ... # read-only |
| __hash__: ClassVar[None] = ... |
| def _CAPICreate(self) -> "Location": ... |
| @staticmethod |
| def callsite(callee: "Location", frames: Sequence["Location"], context: Optional["Context"] = None) -> "Location": ... |
| @staticmethod |
| def file(filename: str, line: int, col: int, context: Optional["Context"] = None) -> "Location": ... |
| @staticmethod |
| def name(name: str, childLoc: Optional["Location"] = None, context: Optional["Context"] = None) -> "Location": ... |
| @staticmethod |
| def unknown(context: Optional["Context"] = None) -> Any: ... |
| def __enter__(self) -> "Location": ... |
| @overload |
| def __eq__(self, arg0: "Location") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __exit__(self, arg0: object, arg1: object, arg2: object) -> None: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> "Context": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class MemRefType(ShapedType): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def affine_map(self) -> AffineMap: ... |
| @property |
| def layout(self) -> Attribute: ... |
| @property |
| def memory_space(self) -> Attribute: ... |
| |
| class Module: |
| def _CAPICreate(self) -> object: ... |
| def create(loc: Optional["Location"] = None) -> "Module": ... |
| def dump(self) -> None: ... |
| @staticmethod |
| def parse(asm: str, context: Optional[Context] = None) -> "Module": ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def body(self) -> "Block": ... |
| @property |
| def context(self) -> object: ... |
| @property |
| def operation(self) -> "_OperationBase": ... |
| |
| class NamedAttribute: |
| @property |
| def attr(self) -> "Attribute": ... |
| @property |
| def name(self) -> str: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class NoneType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| class OpAttributeMap: |
| def __contains__(self, arg0: str) -> bool: ... |
| def __delitem__(self, arg0: str) -> None: ... |
| @overload |
| def __getitem__(self, arg0: str) -> "Attribute": ... |
| @overload |
| def __getitem__(self, arg0: int) -> "NamedAttribute": ... |
| def __len__(self) -> int: ... |
| def __setitem__(self, arg0: str, arg1: "Attribute") -> None: ... |
| |
| class OpOperandList: |
| def __add__(self, arg0: "OpOperandList") -> List[Value]: ... |
| @overload |
| def __getitem__(self, arg0: int) -> Value: ... |
| @overload |
| def __getitem__(self, arg0: slice) -> "OpOperandList": ... |
| def __len__(self) -> int: ... |
| def __setitem__(self, arg0: int, arg1: Value) -> None: ... |
| |
| class OpResult(Value): |
| def __init__(self, value: Value) -> None: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def owner(self) -> "_OperationBase": ... |
| @property |
| def result_number(self) -> int: ... |
| |
| class OpResultList: |
| def __add__(self, arg0: "OpResultList") -> List["OpResult"]: ... |
| @overload |
| def __getitem__(self, arg0: int) -> "OpResult": ... |
| @overload |
| def __getitem__(self, arg0: slice) -> "OpResultList": ... |
| def __len__(self) -> int: ... |
| @property |
| def types(self) -> List["Type"]: ... |
| |
| class OpView(_OperationBase): |
| _ODS_OPERAND_SEGMENTS: ClassVar[None] = ... |
| _ODS_REGIONS: ClassVar[tuple] = ... |
| _ODS_RESULT_SEGMENTS: ClassVar[None] = ... |
| def __init__(self, operation: "_OperationBase") -> None: ... |
| @classmethod |
| def build_generic(cls, results: Optional[Sequence["Type"]] = None, |
| operands: Optional[Sequence["Value"]] = None, |
| attributes: Optional[Dict[str, "Attribute"]] = None, |
| successors: Optional[Sequence["Block"]] = None, |
| regions: Optional[int] = None, |
| loc: Optional["Location"] = None, |
| ip: Optional["InsertionPoint"] = None) -> "_OperationBase": ... |
| @property |
| def context(self) -> "Context": ... |
| @property |
| def operation(self) -> "_OperationBase": ... |
| |
| class Operation(_OperationBase): |
| def _CAPICreate(self) -> object: ... |
| @staticmethod |
| def create(name: str, results: Optional[Sequence["Type"]] = None, |
| operands: Optional[Sequence["Value"]] = None, |
| attributes: Optional[Dict[str, "Attribute"]] = None, |
| successors: Optional[Sequence["Block"]] = None, |
| regions: int = 0, |
| loc: Optional["Location"] = None, |
| ip: Optional["InsertionPoint"] = None) -> "_OperationBase": ... |
| def erase(self) -> None: ... |
| @property |
| def _CAPIPtr(self) -> object: ... |
| @property |
| def context(self) -> "Context": ... |
| @property |
| def name(self) -> str: ... |
| @property |
| def opview(self) -> "OpView": ... |
| @property |
| def parent(self) -> Optional["_OperationBase"]: ... |
| |
| class OperationIterator: |
| def __iter__(self) -> "OperationIterator": ... |
| def __next__(self) -> "OpView": ... |
| |
| class OperationList: |
| def __getitem__(self, arg0: int) -> "OpView": ... |
| def __iter__(self) -> "OperationIterator": ... |
| def __len__(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class RankedTensorType(ShapedType): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def encoding(self) -> Optional[Attribute]: ... |
| |
| class Region: |
| __hash__: ClassVar[None] = ... |
| @overload |
| def __eq__(self, arg0: "Region") -> bool: ... |
| @overload |
| def __eq__(self, arg0: object) -> bool: ... |
| def __iter__(self) -> "BlockIterator": ... |
| @property |
| def blocks(self) -> "BlockList": ... |
| @property |
| def owner(self) -> "OpView": ... |
| |
| class RegionIterator: |
| def __iter__(self) -> "RegionIterator": ... |
| def __next__(self) -> "Region": ... |
| |
| class RegionSequence: |
| def __getitem__(self, arg0: int) -> "Region": ... |
| def __len__(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class ShapedType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get_dim_size(self, dim: int) -> int: ... |
| def is_dynamic_dim(self, dim: int) -> bool: ... |
| def is_dynamic_size(self, *args, **kwargs) -> Any: ... |
| def is_dynamic_stride_or_offset(self, dim_size: int) -> bool: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def element_type(self) -> Type: ... |
| @property |
| def has_rank(self) -> bool: ... |
| @property |
| def has_static_shape(self) -> bool: ... |
| @property |
| def rank(self) -> int: ... |
| @property |
| def shape(self) -> List[int]: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class StringAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def get_typed(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> str: ... |
| |
| class SymbolTable: |
| def __init__(self, arg0: _OperationBase) -> None: ... |
| def erase(self, operation: _OperationBase) -> None: ... |
| @staticmethod |
| def get_symbol_name(symbol: _OperationBase) -> Attribute: ... |
| @staticmethod |
| def get_visibility(symbol: _OperationBase) -> Attribute: ... |
| def insert(self, operation: _OperationBase) -> Attribute: ... |
| @staticmethod |
| def replace_all_symbol_uses(old_symbol: str, new_symbol: str, from_op: _OperationBase) -> None: ... |
| @staticmethod |
| def set_symbol_name(symbol: _OperationBase, name: str) -> None: ... |
| @staticmethod |
| def set_visibility(symbol: _OperationBase, visibility: str) -> None: ... |
| @staticmethod |
| def walk_symbol_tables(from_op: _OperationBase, all_sym_uses_visible: bool, callback: Callable[[_OperationBase, bool], None) -> None: ... |
| def __contains__(self, arg0: str) -> bool: ... |
| def __delitem__(self, arg0: str) -> None: ... |
| def __getitem__(self, arg0: str) -> "OpView": ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class TupleType(Type): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get_tuple(self, *args, **kwargs) -> Any: ... |
| def get_type(self, pos: int) -> Type: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def num_types(self) -> int: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class TypeAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| @property |
| def value(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class UnitAttr(Attribute): |
| def __init__(self, cast_from_attr: Attribute) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def type(self) -> Type: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class UnrankedMemRefType(ShapedType): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| @property |
| def memory_space(self) -> Attribute: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class UnrankedTensorType(ShapedType): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| # TODO: Auto-generated. Audit and fix. |
| class VectorType(ShapedType): |
| def __init__(self, cast_from_type: Type) -> None: ... |
| def get(self, *args, **kwargs) -> Any: ... |
| def isinstance(self, *args, **kwargs) -> Any: ... |
| |
| class _GlobalDebug: |
| flag: ClassVar[bool] = ... |