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

!5600 [MS][LITE][Develop]optimize argmax 

Merge pull request !5600 from chenjianping/lite_dev2
This commit is contained in:
mindspore-ci-bot 2020-09-01 10:46:19 +08:00 committed by Gitee
parent 340bb4c28f 097db73d92
commit 05b8efc805
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/nnacl/arg_min_max.c
View File

@ -87,7 +87,7 @@ void ArgMinMaxTopknFp32(const float *input, float *output, const int *in_shape,
}
void ArgMinMax(const void *input, void *output, const int *in_shape, ArgMinMaxParameter *param) {
if (param->topk_ == 1 && !param->keep_dims_) {
if (param->topk_ == 1) {
ArgMinMaxTopk1(input, output, in_shape, param);
return;
}