OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

pr to master #8

Open
m7grui4p8 wants to merge 201 commits from p69201753/mindspore:cpu-kernel-reuse-1 into master
pull from: p69201753/mindspore:cpu-kernel-reuse-1
merge into: OSSInnovation:master
Conversation 0 Commits 201 Files Changed 306 +4 -8
4 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit b863324d90 - Show all commits

3
tests/st/ops/ascend/test_compoite_random_ops/test_gamma.py
View File

@ -22,7 +22,7 @@ from mindspore.ops import composite as C
from mindspore.common import set_seed
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
set_seed(20)
class Net(nn.Cell):
def __init__(self, shape, seed=0):
@ -31,7 +31,6 @@ class Net(nn.Cell):
self.seed = seed
def construct(self, alpha, beta):
set_seed(20)
return C.gamma(self.shape, alpha, beta, self.seed)

3
tests/st/ops/ascend/test_compoite_random_ops/test_normal.py
View File

@ -22,7 +22,7 @@ from mindspore.ops import composite as C
from mindspore.common import set_seed
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
set_seed(20)
class Net(nn.Cell):
def __init__(self, shape, seed=0):
@ -31,7 +31,6 @@ class Net(nn.Cell):
self.seed = seed
def construct(self, mean, stddev):
set_seed(20)
return C.normal(self.shape, mean, stddev, self.seed)

3
tests/st/ops/ascend/test_compoite_random_ops/test_poisson.py
View File

@ -22,7 +22,7 @@ from mindspore.ops import composite as C
from mindspore.common import set_seed
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
set_seed(20)
class Net(nn.Cell):
def __init__(self, shape, seed=0):
@ -31,7 +31,6 @@ class Net(nn.Cell):
self.seed = seed
def construct(self, mean):
set_seed(20)
return C.poisson(self.shape, mean, self.seed)

3
tests/st/ops/ascend/test_compoite_random_ops/test_uniform.py
View File

@ -22,7 +22,7 @@ from mindspore.ops import composite as C
from mindspore.common import set_seed
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
set_seed(20)
class Net(nn.Cell):
def __init__(self, shape, seed=0):
@ -31,7 +31,6 @@ class Net(nn.Cell):
self.seed = seed
def construct(self, minval, maxval):
set_seed(20)
return C.uniform(self.shape, minval, maxval, self.seed)