From 1cb9133bcfd62bb807d486d096ab255479ea5808 Mon Sep 17 00:00:00 2001 From: liangcanli Date: Wed, 18 Jan 2023 09:48:12 +0800 Subject: [PATCH] fix error order for argmin --- mindspore/core/ops/arg_min.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/core/ops/arg_min.cc b/mindspore/core/ops/arg_min.cc index 20e88977484..b5741354546 100644 --- a/mindspore/core/ops/arg_min.cc +++ b/mindspore/core/ops/arg_min.cc @@ -103,7 +103,9 @@ abstract::AbstractBasePtr ArgMinInfer(const abstract::AnalysisEnginePtr &, const const int64_t input_num = 1; (void)CheckAndConvertUtils::CheckInteger("input size", SizeToLong(input_args.size()), kEqual, input_num, primitive->name()); - return abstract::MakeAbstract(ArgMinInferShape(primitive, input_args), ArgMinInferType(primitive, input_args)); + auto type = ArgMinInferType(primitive, input_args); + auto shape = ArgMinInferShape(primitive, input_args); + return abstract::MakeAbstract(shape, type); } using Argmin = ArgMin;