Browse Source

!4598 support axis is None for all and any interface in graph mode

Merge pull request !4598 from zhangbuxue/support_axis_is_None_for_all_and_any_in_graph_mode
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d1ad33676b
2 changed files with 6 additions and 3 deletions
  1. +4
    -1
      mindspore/_extends/parse/standard_method.py
  2. +2
    -2
      tests/ut/python/pipeline/infer/test_interface_all_and_any_of_tensor.py

+ 4
- 1
mindspore/_extends/parse/standard_method.py View File

@@ -44,6 +44,8 @@ def all_(x, axis=(), keep_dims=False):
Tensor, has the same data type as x.
"""

if axis is None:
axis = ()
reduce_all = P.ReduceAll(keep_dims)
return reduce_all(x, axis)

@@ -60,7 +62,8 @@ def any_(x, axis=(), keep_dims=False):
Returns:
Tensor, has the same data type as x.
"""

if axis is None:
axis = ()
reduce_any = P.ReduceAny(keep_dims)
return reduce_any(x, axis)



+ 2
- 2
tests/ut/python/pipeline/infer/test_interface_all_and_any_of_tensor.py View File

@@ -28,8 +28,8 @@ def test_all_and_any_of_tensor_in_graph():
def construct(self, x):
all_ = x.all()
any_ = x.any()
all_0 = x.all(0, True)
any_0 = x.any(0, True)
all_0 = x.all(None, True)
any_0 = x.any(None, True)
return all_, any_, all_0, any_0

net = Net()


Loading…
Cancel
Save