|
|
@@ -35,7 +35,7 @@ class NetWithSparseGatherV2(nn.Cell): |
|
|
def construct(self, indices, label): |
|
|
def construct(self, indices, label): |
|
|
return self.gather(self.weight1, indices, self.axis) + self.weight2 |
|
|
return self.gather(self.weight1, indices, self.axis) + self.weight2 |
|
|
|
|
|
|
|
|
# @pytest.mark.level0 |
|
|
|
|
|
|
|
|
@pytest.mark.level0 |
|
|
@pytest.mark.platform_arm_ascend_training |
|
|
@pytest.mark.platform_arm_ascend_training |
|
|
@pytest.mark.platform_x86_ascend_training |
|
|
@pytest.mark.platform_x86_ascend_training |
|
|
@pytest.mark.env_onecard |
|
|
@pytest.mark.env_onecard |
|
|
@@ -56,7 +56,7 @@ def test_ftrl_net(): |
|
|
[[0.6821311, 0.6821311]], |
|
|
[[0.6821311, 0.6821311]], |
|
|
[[0.6821311, 0.6821311]]])) |
|
|
[[0.6821311, 0.6821311]]])) |
|
|
|
|
|
|
|
|
# @pytest.mark.level0 |
|
|
|
|
|
|
|
|
@pytest.mark.level0 |
|
|
@pytest.mark.platform_arm_ascend_training |
|
|
@pytest.mark.platform_arm_ascend_training |
|
|
@pytest.mark.platform_x86_ascend_training |
|
|
@pytest.mark.platform_x86_ascend_training |
|
|
@pytest.mark.env_onecard |
|
|
@pytest.mark.env_onecard |
|
|
|