mlir/python/mlir/_mlir_libs/_mlir/ir.pyi

# 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] = ...