forked from mindspore-Ecosystem/mindspore
138 lines
4.8 KiB
ReStructuredText
138 lines
4.8 KiB
ReStructuredText
mindspore.ops.TBERegOp
|
||
======================
|
||
|
||
.. py:class:: mindspore.ops.TBERegOp(op_name)
|
||
|
||
注册TBE算子信息的类。
|
||
|
||
**参数：**
|
||
|
||
- **op_name** (str) - 表示算子名称。
|
||
|
||
.. py:method:: async_flag(async_flag=False)
|
||
|
||
定义算子的计算效率，用于表示是否支持异步计算。
|
||
|
||
**参数：**
|
||
|
||
- **async_flag** (bool) - 表示该算子是否异步的标识。默认值：False。
|
||
|
||
.. py:method:: attr(name=None, param_type=None, value_type=None, value=None, default_value=None, **kwargs)
|
||
|
||
注册TBE算子的属性信息。
|
||
|
||
**参数：**
|
||
|
||
- **name** (str) - 表示算子属性的名称。默认值：None。
|
||
- **param_type** (str) - 表示算子属性的参数类型。默认值：None。
|
||
- **value_type** (str) - 表示算子属性的类型。默认值：None。
|
||
- **value** (str) - 表示算子属性的值。默认值：None。
|
||
- **default_value** (str) - 表示算子属性的默认值。默认值：None。
|
||
- **kwargs** (dict) - 表示算子属性的其他信息。
|
||
|
||
.. py:method:: binfile_name(binfile_name)
|
||
|
||
设置算子底层的二进制文件名，此动作可选。
|
||
|
||
**参数：**
|
||
|
||
- **binfile_name** (str) - 表示算子底层的二进制文件名。
|
||
|
||
.. py:method:: compute_cost(compute_cost=10)
|
||
|
||
定义算子的计算效率，即tiling模块中成本模型的值。
|
||
|
||
**参数：**
|
||
|
||
- **compute_cost** (int) - 表示计算成本的值。默认值：10。
|
||
|
||
.. py:method:: dynamic_compile_static(dynamic_compile_static=False)
|
||
|
||
表示算子是否支持动静合一。
|
||
|
||
**参数：**
|
||
|
||
- **dynamic_compile_static** (bool) - 表示静合一的标识。默认值：False。
|
||
|
||
.. py:method:: dynamic_shape(dynamic_shape=False)
|
||
|
||
表示算子是否支持动态shape。
|
||
|
||
**参数：**
|
||
|
||
- **dynamic_shape** (bool) - 表示是否支持动态shape的标识。默认值：False。
|
||
|
||
.. py:method:: input(index=None, name=None, need_compile=None, param_type=None, shape=None, value_depend=None, **kwargs)
|
||
|
||
注册TBE算子的输入信息。
|
||
|
||
**参数：**
|
||
|
||
- **index** (int) - 表示输入的顺序。默认值：None。
|
||
- **name** (str) - 表示当前输入的名称。默认值：None。
|
||
- **need_compile** (bool) - 表示输入是否需要编译。默认值：None。
|
||
- **param_type** (str) - 表示输入的类型。默认值：None。
|
||
- **shape** (str) - 表示输入的shape。默认值：None。
|
||
- **value_depend** (str) - 表示输入是否为常量值。默认值：None。
|
||
- **kwargs** (dict) - 表示输入的其他信息。
|
||
|
||
.. py:method:: is_dynamic_format(is_dynamic_format=False)
|
||
|
||
表示算子是否需要op_select_format函数来动态选择合适的数据格式和数据类型。
|
||
|
||
**参数：**
|
||
|
||
- **is_dynamic_format** (bool) - 表示否需要op_select_format函数来动态选择合适的数据格式和数据类型的标识。默认值：False。
|
||
|
||
.. py:method:: kernel_name(kernel_name)
|
||
|
||
表示算子名称。
|
||
|
||
**参数：**
|
||
|
||
- **kernel_name** (str) - 表示算子名称。
|
||
|
||
.. py:method:: need_check_supported(need_check_supported=False)
|
||
|
||
表示算子是否需要检查支持。
|
||
|
||
**参数：**
|
||
|
||
- **need_check_supported** (bool) - 表示是否需要检查支持的标识。默认值：False。
|
||
|
||
.. py:method:: op_pattern(pattern=None)
|
||
|
||
表示算子支持的行为类型。
|
||
|
||
**参数：**
|
||
|
||
- **pattern** (str) - 表示算子支持的行为类型，如"broadcast"、"reduce"等。默认值：None。
|
||
|
||
.. py:method:: output(index=None, name=None, need_compile=None, param_type=None, shape=None, **kwargs)
|
||
|
||
注册TBE算子的输出信息。
|
||
|
||
**参数：**
|
||
|
||
- **index** (int) - 表示输出的顺序。默认值：None。
|
||
- **name** (str) - 表示输出的名称。默认值：None。
|
||
- **need_compile** (bool) - 表示输出是否需要编译。默认值：None。
|
||
- **param_type** (str) - 表示输出的类型。默认值：None。
|
||
- **shape** (str) - 表示输出的shape。默认值：None。
|
||
- **kwargs** (dict) - 表示输出的其他信息。
|
||
|
||
.. py:method:: partial_flag(partial_flag=True)
|
||
|
||
定义算子的计算效率，用于表示是否支持部分计算。
|
||
|
||
**参数：**
|
||
|
||
- **partial_flag** (bool) - 表示是否支持部分计算。默认值：True。
|
||
|
||
.. py:method:: reshape_type(reshape_type)
|
||
|
||
指定算子的补维方式。
|
||
|
||
**参数：**
|
||
|
||
- **reshape_type** (str) - 指定算子补维方式的值。例如：输入的shape为 :math:`(2, 3)` ，指定reshape_type="CH"，则补维之后的shape为 :math:`(1, 2, 3, 1)` ，即保留CH轴，NW轴补1。 |