From 097db73d9208c03ec74869dd0d3048a957040b37 Mon Sep 17 00:00:00 2001
From: chenjianping <jpc.chen@huawei.com>
Date: Mon, 31 Aug 2020 21:54:33 +0800
Subject: [PATCH] optimize argmax

---
 mindspore/lite/nnacl/arg_min_max.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/nnacl/arg_min_max.c b/mindspore/lite/nnacl/arg_min_max.c
index 2b61884d44..cae540610c 100644
--- a/mindspore/lite/nnacl/arg_min_max.c
+++ b/mindspore/lite/nnacl/arg_min_max.c
@@ -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;
   }