mindspore.ops.AiCPURegOp

View Source On Gitee
class mindspore.ops.AiCPURegOp(op_name)[source]

Class for AiCPU operator information registration.

Parameters

op_name (str) – Name of operator.

Examples

>>> from mindspore.ops import AiCPURegOp, DataType
>>> stack_op_info = AiCPURegOp("Stack") \
...    .fusion_type("OPAQUE") \
...    .attr("axis", "int") \
...    .input(0, "x", "dynamic") \
...    .output(0, "y", "required") \
...    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
...    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
...    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
...    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
...    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
...    .dtype_format(DataType.U16_Default, DataType.U16_Default) \
...    .dtype_format(DataType.U32_Default, DataType.U32_Default) \
...    .dtype_format(DataType.U64_Default, DataType.U64_Default) \
...    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
...    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
...    .dtype_format(DataType.F64_Default, DataType.F64_Default) \
...    .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default) \
...    .get_op_info()
>>>