|
|
|
@ -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 """
|
|
|
|
|
|
|
|
|
|