diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py
index 18b38f543b..3a4d3e3363 100644
--- a/mindspore/ops/operations/array_ops.py
+++ b/mindspore/ops/operations/array_ops.py
@@ -1232,7 +1232,8 @@ class ArgMinWithValue(PrimitiveWithInfer):
         """init ArgMinWithValue"""
         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