mindspore.ops.op_info_register

mindspore.ops.op_info_register(op_info)[源代码]

用于注册算子的装饰器。

Note

‘op_info’应通过json格式的字符串表示算子信息。’op_info’将添加到算子库’oplib’中。

参数:

  • op_info (Union[str, dict]) - json格式的算子信息。

返回:

Function,返回算子信息注册的装饰器。

样例:

>>> from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
>>> abs_op_info = TBERegOp("Abs") \
...    .fusion_type("ELEMWISE") \
...    .async_flag(False) \
...    .binfile_name("abs.so") \
...    .compute_cost(10) \
...    .kernel_name("abs") \
...    .partial_flag(True) \
...    .op_pattern("formatAgnostic") \
...    .input(0, "x", None, "required", None) \
...    .output(0, "y", True, "required", "all") \
...    .dtype_format(DataType.F16_None, DataType.F16_None) \
...    .dtype_format(DataType.F32_None, DataType.F32_None) \
...    .dtype_format(DataType.I32_None, DataType.I32_None) \
...    .get_op_info()
>>>
>>> @op_info_register(abs_op_info)
... def _abs_tbe():
...    return
...