Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48009 argmin报错顺序问题修复 

Merge pull request !48009 from 李良灿/fixw
This commit is contained in:
i-robot 2023-01-30 03:38:11 +00:00 committed by Gitee
parent 232bc67d26 1cb9133bcf
commit 4fe3b2041b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/core/ops/arg_min.cc
View File

@ -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;