Home
last modified time | relevance | path

Searched refs:TypeFn (Results 1 – 18 of 18) sorted by relevance

/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py74 C[D.m, D.n] += TypeFn.cast_unsigned(U, A[D.m, D.k]) * TypeFn.cast_unsigned(
94 TypeFn.cast_signed(U, AZp)) * (TypeFn.cast_signed(U, B[D.k, D.n]) -
131 D.n] += TypeFn.cast_signed(U, A[D.b, D.m, D.k]) * TypeFn.cast_signed(
150 TypeFn.cast_signed(U, AZp)) * (TypeFn.cast_signed(
165 x[D.m] += TypeFn.cast_signed(U, A[D.m, D.n]) * TypeFn.cast_signed(U, y[D.n])
179 x[D.n] += TypeFn.cast_signed(U, y[D.m]) * TypeFn.cast_signed(U, A[D.m, D.n])
193 C[D.b, D.m] += TypeFn.cast_signed(U, A[D.b, D.m, D.k]) * TypeFn.cast_signed(
206 C[None] += TypeFn.cast_signed(U, A[D.m]) * TypeFn.cast_signed(U, B[D.m])
220 O[D.ow] += TypeFn.cast_signed(U, I[D.ow + D.kw]) * TypeFn.cast_signed(
341 D.f] += (TypeFn.cast_signed(
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Dshape_maps_iteration.py27 C[D.m, D.n] += TypeFn.cast_signed(U, A[D.m, D.k]) * TypeFn.cast_signed(
46 C[None] += TypeFn.cast_signed(U, A[D.m]) * TypeFn.cast_signed(U, B[D.m])
69 O[D.o] += TypeFn.cast_signed(U, I[D.o * 2 + D.k])
H A Dassignments.py38 cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
84 pi = TypeFn.cast_signed(T, const(3.1415926535897931))
85 cst42 = TypeFn.cast_signed(T, const(42))
86 cst1000 = TypeFn.cast_signed(T, exp(const(1e+3)))
H A Demit_misc.py20 O[D.m, D.n] = TypeFn.cast_unsigned(F32, const(42)) + TypeFn.cast_unsigned(
26 O[D.m, D.n] = TypeFn.cast_signed(I32, index(D.m)) + TypeFn.cast_signed(
35 cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
H A Dmetadata.py19 C[D.m, D.n] += TypeFn.cast_signed(U, A[D.m, D.k]) * TypeFn.cast_signed(
H A Demit_fill.py16 O[None] = TypeFn.cast_signed(U, value)
20 O[None] = TypeFn.cast_signed(U, I[None])
H A Demit_pooling.py20 cast=TypeFnAttrDef(default=TypeFn.cast_signed),
70 cast=TypeFn.cast_unsigned,
119 cast=TypeFn.cast_unsigned,
H A Demit_matmul.py28 cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
87 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
139 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
H A Demit_convolution.py22 O[D.n, D.oh, D.ow, D.c] += TypeFn.cast_signed(
24 D.c]) * TypeFn.cast_signed(U, K[D.kh, D.kw, D.c])
H A Darguments.py37 cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md60 C[D.m, D.n] += TypeFn.cast_signed(
61 U, A[D.m, D.k]) * TypeFn.cast_signed(U, B[D.k, D.n])
168 O[D.n, D.oh, D.ow, D.c] += TypeFn.cast_signed(U,
216 * `TypeFn.cast_signed(TypeVar, operand)`
217 * `TypeFn.cast_unsigned(TypeVar, operand)`
221 (`TypeFn.cast_unsigned`) values.
243 cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
254 fun=BinaryFn.mul, cast=TypeFn.cast_unsigned)
275 `TypeFn.cast_signed` or `TypeFn.cast_unsigned` allows for a limited form of
285 signed, while `TypeFn.cast_unsigned` treats them as unsigned.
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgBase.td88 def TypeFn : I32EnumAttr<"TypeFn", "", [
102 def TypeFnAttr : EnumAttr<Linalg_Dialect, TypeFn, "type_fn"> {
/llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml6 # cast=TypeFnAttrDef(default=TypeFn.cast_signed)):
73 # ODS-NEXT: DefaultValuedAttr<TypeFnAttr, "TypeFn::cast_signed">:$cast
90 # IMPL: TypeFn castVal = TypeFn::cast_signed;
200 # O[None] = TypeFn.cast_signed(U, value)
333 # O[None] = TypeFn.cast(U, value)
/llvm-project-15.0.7/mlir/test/python/integration/dialects/linalg/
H A Dopsrun.py300 linalg.matmul(lhs, rhs, outs=[out], cast=TypeFn.cast_unsigned)
338 lhs, rhs, outs=[out], cast=TypeFn.cast_unsigned, emit_generic=True)
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/
H A Dops.py116 cast=TypeFn.cast_unsigned)
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp383 Value buildTypeFn(TypeFn typeFn, Type toType, Value operand) { in buildTypeFn()
385 case TypeFn::cast_signed: in buildTypeFn()
387 case TypeFn::cast_unsigned: in buildTypeFn()
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco.py807 int_value = lang.TypeFn.cast_unsigned(lang.TV.I64, index)
808 return lang.TypeFn.cast_unsigned(lang.T, int_value)
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dcomprehension.py337 class TypeFn: class