mlir.dialects._linalg_enum_gen¶

Attributes¶

_ods_ir

Classes¶

`BinaryFn`	allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9
`ElementwiseArityGroup`	allowed 32-bit signless integer cases: 1, 2, 3
`ElementwiseCaseLimits`	allowed 32-bit signless integer cases:
`ElementwiseKind`	allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23
`IteratorType`	Iterator type
`TernaryFn`	allowed 32-bit signless integer cases: 0
`TypeFn`	allowed 32-bit signless integer cases: 0, 1
`UnaryFn`	allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12
`WinogradConv2DFmr`	allowed 32-bit signless integer cases: 0, 1, 2

Functions¶

`_binaryfn`(x, context)
`_elementwisearitygroup`(x, context)
`_elementwisecaselimits`(x, context)
`_elementwisekind`(x, context)
`_iteratortype`(x, context)
`_ternaryfn`(x, context)
`_typefn`(x, context)
`_unaryfn`(x, context)
`_winogradconv2dfmr`(x, context)
`_binaryfnattr`(x, context)
`_elementwisekindattr`(x, context)
`_iteratortypeenum`(x, context)
`_ternaryfnattr`(x, context)
`_typefnattr`(x, context)
`_unaryfnattr`(x, context)

Module Contents¶

mlir.dialects._linalg_enum_gen._ods_ir¶

class mlir.dialects._linalg_enum_gen.BinaryFn¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9

add = 0¶

sub = 1¶

mul = 2¶

div = 3¶

div_unsigned = 4¶

max_signed = 5¶

min_signed = 6¶

max_unsigned = 7¶

min_unsigned = 8¶

powf = 9¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._binaryfn(x, context)¶

class mlir.dialects._linalg_enum_gen.ElementwiseArityGroup¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 1, 2, 3

Unary = 1¶

Binary = 2¶

Ternary = 3¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._elementwisearitygroup(x, context)¶

class mlir.dialects._linalg_enum_gen.ElementwiseCaseLimits¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases:

LastUnary = 13¶

LastBinary = 23¶

LastTernary = 24¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._elementwisecaselimits(x, context)¶

class mlir.dialects._linalg_enum_gen.ElementwiseKind¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23

exp = 0¶

log = 1¶

abs = 2¶

ceil = 3¶

floor = 4¶

negf = 5¶

reciprocal = 6¶

round = 7¶

sqrt = 8¶

rsqrt = 9¶

square = 10¶

tanh = 11¶

erf = 12¶

add = 13¶

sub = 14¶

mul = 15¶

div = 16¶

div_unsigned = 17¶

max_signed = 18¶

min_signed = 19¶

max_unsigned = 20¶

min_unsigned = 21¶

powf = 22¶

select = 23¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._elementwisekind(x, context)¶

class mlir.dialects._linalg_enum_gen.IteratorType¶

Bases: enum.IntEnum

Iterator type

parallel = 0¶

reduction = 1¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._iteratortype(x, context)¶

class mlir.dialects._linalg_enum_gen.TernaryFn¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0

select = 0¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._ternaryfn(x, context)¶

class mlir.dialects._linalg_enum_gen.TypeFn¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1

cast_signed = 0¶

cast_unsigned = 1¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._typefn(x, context)¶

class mlir.dialects._linalg_enum_gen.UnaryFn¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12

exp = 0¶

log = 1¶

abs = 2¶

ceil = 3¶

floor = 4¶

negf = 5¶

reciprocal = 6¶

round = 7¶

sqrt = 8¶

rsqrt = 9¶

square = 10¶

tanh = 11¶

erf = 12¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._unaryfn(x, context)¶

class mlir.dialects._linalg_enum_gen.WinogradConv2DFmr¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1, 2

F_2_3 = 0¶

F_4_3 = 1¶

F_2_5 = 2¶

__str__()¶: Return str(self).

mlir.dialects._linalg_enum_gen._winogradconv2dfmr(x, context)¶

mlir.dialects._linalg_enum_gen._binaryfnattr(x, context)¶

mlir.dialects._linalg_enum_gen._elementwisekindattr(x, context)¶

mlir.dialects._linalg_enum_gen._iteratortypeenum(x, context)¶

mlir.dialects._linalg_enum_gen._ternaryfnattr(x, context)¶

mlir.dialects._linalg_enum_gen._typefnattr(x, context)¶

mlir.dialects._linalg_enum_gen._unaryfnattr(x, context)¶