Searched refs:cast_unsigned (Results 1 – 13 of 13) sorted by relevance
| /llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/ |
| H A D | emit_pooling.py | 70 cast=TypeFn.cast_unsigned, 119 cast=TypeFn.cast_unsigned,
|
| H A D | emit_misc.py | 20 O[D.m, D.n] = TypeFn.cast_unsigned(F32, const(42)) + TypeFn.cast_unsigned(
|
| H A D | emit_matmul.py | 87 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned) 139 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
|
| /llvm-project-15.0.7/mlir/docs/Dialects/Linalg/ |
| H A D | OpDSL.md | 217 * `TypeFn.cast_unsigned(TypeVar, operand)` 221 (`TypeFn.cast_unsigned`) values. 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. 291 * cast_unsigned(I32 -> I64) -> `arith.ExtUIOp` 292 * cast_unsigned(F32 -> I32) -> `arith.FPToUIOp`
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgBase.td | 90 I32EnumAttrCase<"cast_unsigned", 1>
|
| H A D | LinalgNamedStructuredOps.yaml | 285 fn_name: cast_unsigned 293 fn_name: cast_unsigned 3072 fn_name: cast_unsigned 3309 fn_name: cast_unsigned
|
| /llvm-project-15.0.7/mlir/test/python/integration/dialects/linalg/ |
| H A D | opsrun.py | 300 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 D | ops.py | 116 cast=TypeFn.cast_unsigned)
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 74 C[D.m, D.n] += TypeFn.cast_unsigned(U, A[D.m, D.k]) * TypeFn.cast_unsigned( 745 D.c] = ReduceFn.max_unsigned[D.kh, D.kw](TypeFn.cast_unsigned( 804 D.c] = ReduceFn.min_unsigned[D.kh, D.kw](TypeFn.cast_unsigned(
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 807 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 D | comprehension.py | 349 cast_unsigned = TypeFnType("cast_unsigned") variable in TypeFn
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | generalize-named-polymorphic-ops.mlir | 44 %0 = linalg.matmul {cast = #linalg.type_fn<cast_unsigned>}
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 387 case TypeFn::cast_unsigned: in buildTypeFn()
|