Searched refs:op_class_name (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | dsl.py | 130 dsl_func=None, *, op_name=None, op_class_name=None argument
135 linalg_structured_op, op_name=op_name, op_class_name=op_class_name
140 if op_class_name is None:
142 op_class_name = f"{''.join(x.title() for x in op_name.split('_'))}Op"
145 name=op_name, cpp_class_name=op_class_name, doc=inspect.getdoc(dsl_func)
|
| H A D | emitter.py | 281 op_class_name: str,
305 or not op_class_name in linalg.__dict__.keys()
312 named_op = getattr(linalg, op_class_name)(result_types, ins, outs)
|
| /llvm-project/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 99 @linalg_structured_op(op_class_name="NegFOp")
111 @linalg_structured_op(op_class_name="ReciprocalOp")
345 @linalg_structured_op(op_class_name="PowFOp")
|