Searched refs:ReduceFn (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/docs/Dialects/Linalg/ |
| H A D | OpDSL.md | 204 * `ReduceFn.add` (also overloading the inplace `+=` on a LHS)
205 * `ReduceFn.mul`
206 * `ReduceFn.max_signed`
207 * `ReduceFn.min_signed`
208 * `ReduceFn.max_unsigned`
209 * `ReduceFn.min_unsigned`
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 722 O[D.n, D.oh, D.ow, D.c] = ReduceFn.max_signed[D.kh, D.kw](TypeFn.cast_signed(
745 D.c] = ReduceFn.max_unsigned[D.kh, D.kw](TypeFn.cast_unsigned(
763 O[D.n, D.c, D.oh, D.ow] = ReduceFn.max_signed[D.kh, D.kw](TypeFn.cast_signed(
781 O[D.n, D.oh, D.ow, D.c] = ReduceFn.min_signed[D.kh, D.kw](TypeFn.cast_signed(
804 D.c] = ReduceFn.min_unsigned[D.kh, D.kw](TypeFn.cast_unsigned(
858 D.c] = ReduceFn.max_signed[D.kd, D.kh, D.kw](TypeFn.cast_signed(
886 D.c] = ReduceFn.min_signed[D.kd, D.kh, D.kw](TypeFn.cast_signed(
|
| /llvm-project-15.0.7/clang/lib/CodeGen/ |
| H A D | CGOpenMPRuntimeGPU.cpp | 2397 QualType ReductionArrayTy, llvm::Function *ReduceFn, SourceLocation Loc) { in emitShuffleAndReduceFunction() argument
2514 CGF, Loc, ReduceFn, {LocalReduceListPtr, RemoteReduceListPtr}); in emitShuffleAndReduceFunction()
2671 llvm::Function *ReduceFn) { in emitListToGlobalReduceFunction() argument
2751 CGF, Loc, ReduceFn, {GlobalReduceList, ReducedPtr}); in emitListToGlobalReduceFunction()
2881 llvm::Function *ReduceFn) { in emitGlobalToListReduceFunction() argument
2961 CGF, Loc, ReduceFn, {ReducedPtr, GlobalReduceList}); in emitGlobalToListReduceFunction()
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | comprehension.py | 393 class ReduceFn: class
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 571 value = lang.ReduceFn.add[reduce_dims](value)
|