From: @bairongz
Reviewed-by: @zh_qh,@zhunaipan
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2020-12-11 15:29:33 +08:00 committed by Gitee
commit 10d326ee62
3 changed files with 1 additions and 14 deletions

View File

@ -99,7 +99,7 @@ class ReduceLogSumExp(Cell):
class Range(Cell):
r"""
Creates a sequence of numbers.
Creates a sequence of numbers in range [start, limit) with step size delta.
Args:
start (Union[int, float]): If `limit` is `None`, the value acts as limit in the range and first entry

View File

@ -23,11 +23,6 @@ from . import operations as P
from .operations import _grad_ops
def arange(start, limit=None, delta=1):
from mindspore.nn import Range
return Range(start, limit, delta)()
typeof = Primitive('typeof')
hastype = Primitive('hastype')
cast = P.Cast()

View File

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