mindspore.ops.TBERegOp

class mindspore.ops.TBERegOp(op_name)

Class for TBE operator information register.

Parameters

op_name (str) – kernel name.

Examples

>>> 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()
>>>

async_flag(async_flag=False)

Define the calculation efficiency of the operator, whether the asynchronous calculation is supported.

Parameters

async_flag (bool) – Value of async flag. Default: false.

attr(name=None, param_type=None, value_type=None, value=None, default_value=None, **kwargs)

Register TBE op attribute information.

Parameters

• name (str) – Name of the attribute. Default: None.

• param_type (str) – Param type of the attribute. Default: None.

• value_type (str) – Type of the attribute. Default: None.

• value (str) – Value of the attribute. Default: None.

• default_value (str) – Default value of attribute. Default: None.

• kwargs (dict) – Other information of the attribute.

binfile_name(binfile_name)

Set the binary file name of the operator, it is optional.

Parameters

binfile_name (str) – The binary file name of the operator.

compute_cost(compute_cost=10)

Define the calculation efficiency of operator, which refers to the value of the cost model in the tiling module.

Parameters

compute_cost (int) – Value of compute cost. Default: 10.

dynamic_compile_static(dynamic_compile_static=False)

Whether the operator supports dynamic compile static.

Parameters

dynamic_compile_static (bool) – Value of dynamic compile static. Default: false.

dynamic_shape(dynamic_shape=False)

Whether the operator supports dynamic shape.

Parameters

dynamic_shape (bool) – Value of dynamic shape. Default: false.

input(index=None, name=None, need_compile=None, param_type=None, shape=None, value_depend=None, **kwargs)

Register TBE op input information.

Parameters

• index (int) – Order of the input. Default: None.

• name (str) – Name of the input. Default: None.

• need_compile (bool) – Whether the input needs to be compiled or not. Default: None.

• param_type (str) – Type of the input. Default: None.

• shape (str) – Shape of the input. Default: None.

• value_depend (str) – Whether the input is constant value depend. Default: None.

• kwargs (dict) – Other information of the input.

is_dynamic_format(is_dynamic_format=False)

Whether the operator need calop_select_format api.

Parameters

is_dynamic_format (bool) – Value of is_dynamic_format. Default: false.

kernel_name(kernel_name)

The name of operator kernel.

Parameters

kernel_name (str) – Name of operator kernel.

need_check_supported(need_check_supported=False)

Whether the operator need check supports.

Parameters

need_check_supported (bool) – Value of need_check_supported. Default: false.

op_pattern(pattern=None)

The behavior type of operator, such as broadcast, reduce and so on.

Parameters

pattern (str) – Value of op pattern.

output(index=None, name=None, need_compile=None, param_type=None, shape=None, **kwargs)

Register TBE op output information.

Parameters

• index (int) – Order of the output. Default: None.

• name (str) – Name of the output. Default: None.

• need_compile (bool) – Whether the output needs to be compiled or not. Default: None.

• param_type (str) – Type of the output. Default: None.

• shape (str) – Shape of the output. Default: None.

• kwargs (dict) – Other information of the output.

partial_flag(partial_flag=True)

Define the calculation efficiency of operator, whether the partial calculation is supported.

Parameters

partial_flag (bool) – Value of partial flag. Default: true.

reshape_type(reshape_type)

Reshape type of operator.

Parameters

reshape_type (str) – Value of reshape type.