Fix ReduceMax op bool type problem.

5 years ago · 618c2fd523
parent a5e8c1eab3
commit 618c2fd523
1 changed files with 2 additions and 0 deletions
--- a/mindspore/ops/operations/math_ops.py
+++ b/mindspore/ops/operations/math_ops.py
@ -497,6 +497,8 @@ class ReduceMax(_Reduce):
        super(ReduceMax, self).__init__(keep_dims)
        self.__setattr_flag__ = True

+    def __infer__(self, input_x, axis):
+        return self.do_infer(input_x, axis, mstype.number_type + (mstype.bool_,))

 class ReduceMin(_Reduce):
    """