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

!9417 Mod test_dense in ut 

From: @wanyiming
Reviewed-by: @zh_qh,@kingxian
Signed-off-by: @kingxian
This commit is contained in:
mindspore-ci-bot 2020-12-04 11:38:47 +08:00 committed by Gitee
parent 493594e045 5ab8007475
commit 9466429fdd
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/ops/ascend/test_dense.py
View File

@ -35,7 +35,7 @@ class MultiLayerDense(nn.Cell):
def __init__(self):
super(MultiLayerDense, self).__init__()
self.dense1 = nn.Dense(in_channels=256, out_channels=512)
self.dense1 = nn.Dense(in_channels=512, out_channels=1024)
self.dense2 = nn.Dense(in_channels=512, out_channels=1024)
@ms_function
def construct(self, x):

4
tests/st/ops/gpu/test_dense.py
View File

@ -1,4 +1,4 @@
# Copyright 2019 Huawei Technologies Co., Ltd
# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@ -33,7 +33,7 @@ class MultiLayerDense(nn.Cell):
def __init__(self):
super(MultiLayerDense, self).__init__()
self.dense1 = nn.Dense(in_channels=256, out_channels=512)
self.dense1 = nn.Dense(in_channels=512, out_channels=1024)
self.dense2 = nn.Dense(in_channels=512, out_channels=1024)
def construct(self, x):
x = self.dense1(x)