Browse Source

fix codedex

tags/v0.3.0-alpha
zjun 6 years ago
parent
commit
3f8dfb70e2
1 changed files with 18 additions and 18 deletions
  1. +18
    -18
      mindspore/ops/op_info_register.py

+ 18
- 18
mindspore/ops/op_info_register.py View File

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


Loading…
Cancel
Save