p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!9631 fix graph kernel doc problem. 

From: @zengzitao
Reviewed-by: @ckey_dou,@gaoxiong1,@ckey_dou
Signed-off-by: @ckey_dou
This commit is contained in:
mindspore-ci-bot 2020-12-08 16:12:51 +08:00 committed by Gitee
parent 25e099be92 517268e816
commit 0f18cdca84
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/nn/cell.py
View File

@ -690,6 +690,7 @@ class Cell(Cell_):
Dict[Parameter, Parameter], returns a dict of original parameter and replaced parameter. Dict[Parameter, Parameter], returns a dict of original parameter and replaced parameter.
""" """
replace = dict() replace = dict()
def _updata(param): def _updata(param):
if param in replace: if param in replace:
return replace[param] return replace[param]
@ -1078,6 +1079,10 @@ class GraphKernel(Cell):
A `GraphKernel` a composite of basic primitives and can be compiled into a fused kernel automatically when A `GraphKernel` a composite of basic primitives and can be compiled into a fused kernel automatically when
enable_graph_kernel in context is set to True. enable_graph_kernel in context is set to True.
Args:
auto_prefix (bool): Recursively generate namespaces. Default: True.
flags (dict) : Set graph flags. Default: None.
Examples: Examples:
>>> class Relu(nn.GraphKernel): >>> class Relu(nn.GraphKernel):
... def __init__(self): ... def __init__(self):
@ -1088,8 +1093,8 @@ class GraphKernel(Cell):
... return self.max(P.Fill()(P.DType()(x), P.Shape()(x), 0.0), x) ... return self.max(P.Fill()(P.DType()(x), P.Shape()(x), 0.0), x)
""" """
def __init__(self, auto_prefix=True, pips=None): def __init__(self, auto_prefix=True, flags=None):
super(GraphKernel, self).__init__(auto_prefix, pips) super(GraphKernel, self).__init__(auto_prefix, flags)
class_name = self.__class__.__name__ class_name = self.__class__.__name__
self.add_flags(graph_kernel=class_name) self.add_flags(graph_kernel=class_name)