From 32519ef5704f4929fec2ebd25d1cf026568e87b4 Mon Sep 17 00:00:00 2001 From: fangzehua Date: Mon, 27 Jul 2020 16:53:12 +0800 Subject: [PATCH] fix argmaxwithvalue --- mindspore/ops/operations/array_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 8a8ad5ea59..5d1b348feb 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1185,7 +1185,8 @@ class ArgMaxWithValue(PrimitiveWithInfer): """init ArgMaxWithValue""" self.axis = axis self.keep_dims = keep_dims - _check_infer_attr_reduce(axis, keep_dims, self.name) + validator.check_value_type('keep_dims', keep_dims, [bool], self.name) + validator.check_value_type('axis', axis, [int], self.name) def infer_shape(self, x_shape): axis = self.axis