From 5ab8007475d633c1bf945a28e5b23eb59d361a9c Mon Sep 17 00:00:00 2001 From: wanyiming Date: Thu, 3 Dec 2020 11:32:03 +0800 Subject: [PATCH] mod dense --- tests/st/ops/ascend/test_dense.py | 2 +- tests/st/ops/gpu/test_dense.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/st/ops/ascend/test_dense.py b/tests/st/ops/ascend/test_dense.py index acb90b362e..d696de7d42 100644 --- a/tests/st/ops/ascend/test_dense.py +++ b/tests/st/ops/ascend/test_dense.py @@ -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): diff --git a/tests/st/ops/gpu/test_dense.py b/tests/st/ops/gpu/test_dense.py index 0b69869266..99f23c3972 100644 --- a/tests/st/ops/gpu/test_dense.py +++ b/tests/st/ops/gpu/test_dense.py @@ -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)