提交 1dc7bdb6 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!863 Modify operator information description

Merge pull request !863 from zjun/fix_codedex
......@@ -156,7 +156,7 @@ class RegOp:
def fusion_type(self, fusion_type):
"""
Register fusion type.
Fusion type of the operator.
Args:
fusion_type (str): Value of fusion type.
......@@ -167,7 +167,7 @@ class RegOp:
def dtype_format(self, *args):
"""
Register dtype and format.
A dtype and format supported by the operator.
Args:
args (tuple): Value of dtype and format.
......@@ -206,7 +206,7 @@ class RegOp:
class AkgRegOp(RegOp):
"""Class for Akg op info register"""
"""Class for Akg op info register."""
def __init__(self, op_name):
super(AkgRegOp, self).__init__(op_name)
......@@ -325,7 +325,7 @@ class AiCPURegOp(RegOp):
class TBERegOp(RegOp):
"""Class for TBE op info register."""
def __init__(self, op_name=""):
def __init__(self, op_name):
super(TBERegOp, self).__init__(op_name)
self.imply_type = "TBE"
self.async_flag_ = False
......@@ -339,10 +339,10 @@ class TBERegOp(RegOp):
def async_flag(self, async_flag):
"""
Register async flag.
Define the calculation efficiency of operator, whether to support asynchronous calculation.
Args:
async_flag (bool): Value of async flag.
async_flag (bool): Value of async flag. Default: false.
"""
self._is_bool(async_flag)
self.async_flag_ = async_flag
......@@ -350,10 +350,10 @@ class TBERegOp(RegOp):
def binfile_name(self, binfile_name):
"""
Register binfile name.
Binary file name of operator. The option is optional.
Args:
binfile_name (str): Name of op binfile.
binfile_name (str): File name of operator binary.
"""
self._is_string(binfile_name)
self.binfile_name_ = binfile_name
......@@ -361,10 +361,10 @@ class TBERegOp(RegOp):
def compute_cost(self, compute_cost):
"""
Register compute cost.
Define the calculation efficiency of operator, which refers to cost model value of the tiling module.
Args:
compute_cost (int): Value of compute cost.
compute_cost (int): Value of compute cost. Default: 10.
"""
self._is_int(compute_cost)
self.compute_cost_ = compute_cost
......@@ -372,10 +372,10 @@ class TBERegOp(RegOp):
def kernel_name(self, kernel_name):
"""
Register kernel name.
The name of operator kernel.
Args:
kernel_name (str): Name of op kernel.
kernel_name (str): Name of operator kernel.
"""
self._is_string(kernel_name)
self.kernel_name_ = kernel_name
......@@ -383,10 +383,10 @@ class TBERegOp(RegOp):
def partial_flag(self, partial_flag):
"""
Register partial flag.
Define the calculation efficiency of operator, whether to support partial calculation.
Args:
partial_flag (bool): Value of partial flag.
partial_flag (bool): Value of partial flag. Default: true.
"""
self._is_bool(partial_flag)
self.partial_flag_ = partial_flag
......@@ -394,7 +394,7 @@ class TBERegOp(RegOp):
def reshape_type(self, reshape_type):
"""
Register reshape type.
Reshape type of operator.
Args:
reshape_type (str): Value of reshape type.
......@@ -405,10 +405,10 @@ class TBERegOp(RegOp):
def dynamic_format(self, dynamic_format):
"""
Register dynamic format.
Whether the operator supports dynamic selection of format and dtype.
Args:
reshape_type (bool): Value of dynamic format.
dynamic_format (bool): Value of dynamic format. Default: false.
"""
self._is_bool(dynamic_format)
self.dynamic_format_ = dynamic_format
......@@ -416,7 +416,7 @@ class TBERegOp(RegOp):
def op_pattern(self, pattern=None):
"""
Register TBE op pattern information.
The behavior type of opeator, such as broadcast, reduce and so on.
Args:
pattern (str): Value of op pattern.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册