|
|
|
@@ -24,7 +24,6 @@ from mindspore import Tensor |
|
|
|
from mindspore.common import dtype as mstype |
|
|
|
from mindspore.ops import composite as C |
|
|
|
from mindspore.ops import operations as P |
|
|
|
from mindspore.ops import functional as F |
|
|
|
from mindspore.ops import prim_attr_register, PrimitiveWithInfer |
|
|
|
from ..ut_filter import non_graph_engine |
|
|
|
from ....mindspore_test_framework.mindspore_test import mindspore_test |
|
|
|
@@ -140,13 +139,6 @@ def test_eye(): |
|
|
|
assert np.all(eye_output.asnumpy() == expect) |
|
|
|
|
|
|
|
|
|
|
|
def test_arange(): |
|
|
|
""" test_arange """ |
|
|
|
F.arange(10) |
|
|
|
F.arange(1, 5) |
|
|
|
F.arange(1, 10, 2) |
|
|
|
|
|
|
|
|
|
|
|
class VirtualLossGrad(PrimitiveWithInfer): |
|
|
|
""" VirtualLossGrad definition """ |
|
|
|
|
|
|
|
|