commit 752e15ace438531c9bfc4c5fc00d917c0c37a558
author H1Gdev <h1g.z.ouroboros+github@gmail.com> Tue Oct 19 15:46:25 2021 +0900
committer H1Gdev <h1g.z.ouroboros+github@gmail.com> Tue Oct 19 15:48:03 2021 +0900
tree 09ee39a4854afa64a59b1f3f577ec660b1fee7ee
parent 89c2727b9de0ab84af383c6699eafde7c2ba3977

[NNAPI]Fix error in axis of reduce_mean.

tensorflow/lite/delegates/nnapi/nnapi_delegate.cc

diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
index 7426951..a90c9b2 100644
--- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
+++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
@@ -5445,7 +5445,8 @@
                   reg->builtin_code == kTfLiteBuiltinReduceMax ||
                   reg->builtin_code == kTfLiteBuiltinReduceMin ||
                   reg->builtin_code == kTfLiteBuiltinReduceProd ||
-                  reg->builtin_code == kTfLiteBuiltinSum) &&
+                  reg->builtin_code == kTfLiteBuiltinSum ||
+                  reg->builtin_code == kTfLiteBuiltinMean) &&
                  (input_pos == 1)) {
         // The axis needs, be converted to a tensor if specified as scalar
         const TfLiteTensor& axis_tensor =