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 ...