From 1556ce86ea355f1d60d5665cafd75575131816b9 Mon Sep 17 00:00:00 2001 From: fangzehua Date: Sat, 25 Jul 2020 09:52:23 +0800 Subject: [PATCH] fix argminwith value --- 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 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