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

!9618 Modify the names of parameter check 

From: @peixu_ren
Reviewed-by: @zichun_ye,@sunnybeike
Signed-off-by: @sunnybeike
This commit is contained in:
mindspore-ci-bot 2020-12-08 11:25:26 +08:00 committed by Gitee
parent f6a22cb455 35b4a804e2
commit e7555043bd
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/nn/layer/math.py
View File

@ -245,7 +245,7 @@ class LGamma(Cell):
def construct(self, x):
input_dtype = self.dtype(x)
_check_input_dtype("input", input_dtype, [mstype.float16, mstype.float32], self.cls_name)
_check_input_dtype("x", input_dtype, [mstype.float16, mstype.float32], self.cls_name)
infinity = self.fill(input_dtype, self.shape(x), self.inf)
need_to_reflect = self.less(x, 0.5)
@ -352,7 +352,7 @@ class DiGamma(Cell):
def construct(self, x):
input_dtype = self.dtype(x)
_check_input_dtype("input_x", input_dtype, [mstype.float16, mstype.float32], self.cls_name)
_check_input_dtype("x", input_dtype, [mstype.float16, mstype.float32], self.cls_name)
need_to_reflect = self.less(x, 0.5)
neg_input = -x
z = self.select(need_to_reflect, neg_input, x - 1)
@ -612,8 +612,8 @@ class IGamma(Cell):
def construct(self, a, x):
a_dtype = self.dtype(a)
x_dtype = self.dtype(x)
_check_input_dtype("input_a", a_dtype, [mstype.float32], self.cls_name)
_check_input_dtype("input_x", x_dtype, a_dtype, self.cls_name)
_check_input_dtype("a", a_dtype, [mstype.float32], self.cls_name)
_check_input_dtype("x", x_dtype, a_dtype, self.cls_name)
domain_error = self.logicalor(self.less(x, 0), self.less(a, 0))
use_igammac = self.logicaland(self.greater(x, 1), self.greater(x, a))
ax = a * self.log(x) - x - self.lgamma(a)
@ -688,8 +688,8 @@ class LBeta(Cell):
def construct(self, x, y):
x_dtype = self.dtype(x)
y_dtype = self.dtype(y)
_check_input_dtype("input_x", x_dtype, [mstype.float16, mstype.float32], self.cls_name)
_check_input_dtype("input_y", y_dtype, x_dtype, self.cls_name)
_check_input_dtype("x", x_dtype, [mstype.float16, mstype.float32], self.cls_name)
_check_input_dtype("y", y_dtype, x_dtype, self.cls_name)
x_plus_y = x + y
para_shape = self.shape(x_plus_y)
broadcastto = P.BroadcastTo(para_shape)