Home
last modified time | relevance | path

Searched defs:OpView (Results 1 – 12 of 12) sorted by relevance

/llvm-project/mlir/python/mlir/dialects/linalg/
H A D__init__.py64 transpose( input: Union[Operation, OpView, Sequence[Value]], *, outs: List[Union[Operation, OpView, Sequence[Value]]], permutation: Union[DenseI64ArrayAttr, List[int]], ) global() argument
86 broadcast( input: Union[Operation, OpView, Sequence[Value]], *, outs: List[Union[Operation, OpView, Sequence[Value]]], dimensions: Union[DenseI64ArrayAttr, List[int]], ) global() argument
/llvm-project/mlir/python/mlir/dialects/
H A Dpdl.py31 valueType: Optional[Union[OpView, Operation, Value]] = None,
48 type: Optional[Union[OpView, Operation, Value]] = None,
64 types: Optional[Union[OpView, Operation, Value]] = None,
81 args: Optional[Sequence[Union[OpView, Operation, Value]]] = None,
83 types: Optional[Sequence[Union[OpView, Operation, Value]]] = None,
139 op: Union[OpView, Operation, Value],
141 with_op: Optional[Union[OpView, Operation, Value]] = None,
142 with_values: Optional[Sequence[Union[OpView, Operation, Value]]] = None,
160 parent: Union[OpView, Operation, Value],
177 root: Optional[Union[OpView, Operation, Value]] = None,
[all …]
H A D_ods_common.py91 get_op_result_or_value( arg: _Union[ _cext.ir.OpView, _cext.ir.Operation, _cext.ir.Value, _cext.ir.OpResultList ] ) global() argument
113 get_op_results_or_values( arg: _Union[ _cext.ir.OpView, _cext.ir.Operation, _Sequence[_Union[_cext.ir.OpView, _cext.ir.Operation, _cext.ir.Value]], ] ) global() argument
115 get_op_results_or_values( arg: _Union[ _cext.ir.OpView, _cext.ir.Operation, _Sequence[_Union[_cext.ir.OpView, _cext.ir.Operation, _cext.ir.Value]], ] ) global() argument
133 get_op_result_or_op_results( op: _Union[_cext.ir.OpView, _cext.ir.Operation], ) global() argument
[all...]
H A Dscf.py32 iter_args: Optional[Union[Operation, OpView, Sequence[Value]]] = None, argument
/llvm-project/mlir/python/mlir/dialects/transform/
H A Dgpu.py26 target: Union[Operation, OpView, Value], argument
38 target: Union[Operation, OpView, Value], argument
49 result_type_or_target: Union[Operation, OpView, Type, Value], argument
82 target: Union[Operation, OpView, Value], argument
95 target: Union[Operation, OpView, Value], argument
107 result_type_or_target: Union[Operation, OpView, Value, Type], argument
H A Dmemref.py27 alloca: Union[Operation, OpView, Value], argument
35 def __init__(self, alloca: Union[Operation, OpView, Value], *, loc=None, ip=None): argument
40 get_global_type_or_alloca: Union[Operation, OpView, Type, Value], argument
73 target: Union[Operation, OpView, Value], argument
85 target: Union[Operation, OpView, Value], argument
97 target_or_factor: Union[int, IntegerAttr, Operation, OpView, Value] = None, argument
H A Dstructured.py38 target: Union[Operation, OpView, Value], argument
87 producer_op: Union[Operation, OpView, Value], argument
98 producer_op: Union[Operation, OpView, Value], argument
108 fused_op_type_or_producer_op: Union[Operation, OpView, Type, Value], argument
183 __init__( self, forall_op_type: Type, tiled_op_type: Type, target: Union[Operation, OpView, Value], *, total_num_threads: Union[int, IntegerAttr], desired_bit_alignment: Union[int, IntegerAttr], loc=None, ip=None, ) global() argument
195 __init__( self, target: Union[Operation, OpView, Value], *, total_num_threads: Union[int, IntegerAttr], desired_bit_alignment: Union[int, IntegerAttr], loc=None, ip=None, ) global() argument
206 __init__( self, forall_op_type_or_target: Union[Operation, OpView, Type, Value], tiled_op_type_or_none: Optional[Type] = None, target_or_none: Optional[Union[Operation, OpView, Value]] = None, *, total_num_threads: Union[int, IntegerAttr], desired_bit_alignment: Union[int, IntegerAttr], loc=None, ip=None, ) global() argument
241 __init__( self, target: Union[Operation, OpView, Value], vector_sizes: Optional[Union[DynamicIndexList, ArrayAttr]] = None, *, vectorize_nd_extract: Optional[bool] = None, scalable_sizes: OptionalBoolList = None, static_vector_sizes: OptionalIntList = None, loc=None, ip=None, ) global() argument
375 __init__( self, target: Union[Operation, OpView, Value], *, 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, transpose_paddings: Optional[ Union[ArrayAttr, Sequence[Union[ArrayAttr, IntOrAttrList]]] ] = None, copy_back_op: Optional[Union[str, StringAttr]] = None, loc=None, ip=None, ) global() argument
479 __init__( self, target: Union[Operation, Value, OpView], *, sizes: Optional[Union[DynamicIndexList, ArrayAttr]] = None, interchange: OptionalIntList = None, loc=None, ip=None, ) global() argument
491 __init__( self, loop_types_or_target: Union[Type, List[Type], Operation, Value], target_or_none: Optional[Union[Operation, Value, OpView]] = None, *, sizes: Optional[Union[DynamicIndexList, ArrayAttr]] = None, interchange: OptionalIntList = None, loc=None, ip=None, ) global() argument
542 __init__( self, loops_type: Type, tiled_op_type: Type, target: Union[Operation, Value, OpView], *, num_threads: Optional[MixedValues] = None, tile_sizes: MixedValues = None, mapping=None, loc=None, ip=None, ) global() argument
555 __init__( self, target: Union[Operation, Value, OpView], *, num_threads: Optional[MixedValues] = None, tile_sizes: MixedValues = None, mapping=None, loc=None, ip=None, ) global() argument
571 __init__( self, loops_type_or_target: Union[ Type, Union[Operation, Value, OpView] ], tiled_op_type_or_none: Optional[Type] = None, target_or_none: Optional[Union[Operation, Value, OpView]] = None, *, num_threads: MixedValues = None, tile_sizes: MixedValues = None, mapping=None, loc=None, ip=None, ) global() argument
[all...]
H A Dbufferization.py27 target: Union[Operation, OpView, Value], argument
35 def __init__(self, target: Union[Operation, OpView, Value], *, loc=None, ip=None): argument
41 target_or_none: Optional[Union[Operation, OpView, Value]] = None, argument
69 target: Union[Operation, OpView, Value], argument
86 target: Union[Operation, OpView, Value], argument
103 target_or_none: Optional[Union[Operation, OpView, Value]] = None, argument
H A Dtensor.py26 target: Union[Operation, OpView, Value], argument
37 target: Union[Operation, OpView, Value], argument
48 target_or_num_loops: Union[int, IntegerAttr, Operation, OpView, Value] = None, argument
H A D__init__.py41 target: Union[Operation, Value, OpView], argument
126 ] = None,
/llvm-project/mlir/python/mlir/dialects/transform/extras/
H A D__init__.py80 ir.OpView,
82 Sequence[Union[str, ir.OpView]],
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Ddsl.py70 *ins: Union[ir.Operation, ir.OpView, ir.Value],