From 7d23a146f7962ff218d8d4bbad24d651eaafa2c5 Mon Sep 17 00:00:00 2001
From: dayschan <chendeshi@huawei.com>
Date: Mon, 22 Aug 2022 20:27:55 +0800
Subject: [PATCH] fix reshape expander: support shape value 0

---
 .../tools/graph_kernel/converter/expanders/reshape.cc    | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mindspore/lite/tools/graph_kernel/converter/expanders/reshape.cc b/mindspore/lite/tools/graph_kernel/converter/expanders/reshape.cc
index d24adc5baa5..9a04456fb16 100644
--- a/mindspore/lite/tools/graph_kernel/converter/expanders/reshape.cc
+++ b/mindspore/lite/tools/graph_kernel/converter/expanders/reshape.cc
@@ -57,6 +57,15 @@ class Reshape : public OpDesc {
       MS_LOG(INFO) << "Reshape's attr shape is neither Tensor nor ValueTuple. Expand failed";
       return {};
     }
+    for (size_t i = 0; i < shape.size(); i++) {
+      if (shape[i] == 0) {
+        if (input_x->shape.size() <= i) {
+          MS_LOG(INFO) << "Reshape's attr shape[" << i << "] is 0, but input's rank is " << input_x->shape.size();
+          return {};
+        }
+        shape[i] = input_x->shape[i];
+      }
+    }
     auto result = gb.Reshape(input_x, shape);
     return {result};
   }