diff --git a/docs/user_guide/op_lists.rst b/docs/user_guide/op_lists.rst
index 63a033467c7ac3856aecbff3ed7d9c4ceafeea8e..5e674e8c060c4478e2b77feb61d8eb789c5a278f 100644
--- a/docs/user_guide/op_lists.rst
+++ b/docs/user_guide/op_lists.rst
@@ -37,7 +37,7 @@ Operator lists
     "RELU1","Y",""
     "RELU6","Y",""
     "RELUX","Y",""
-    "RESHAPE","Y","Limited support: GPU is full supported, for CPU only supports softmax-like usage."
+    "RESHAPE","Y","Limited support: GPU only supports softmax-like usage, CPU only supports the usage which not change the storage format."
     "RESIZE_BILINEAR","Y",""
     "RNN","",""
     "RPN_PROPOSAL_LAYER","Y",""
diff --git a/mace/core/net.cc b/mace/core/net.cc
index 4922bc9470b388d76304a4b7e08362c9226668ed..8a9c121aa864bd66a21ae6a1dade49466aaa1bd2 100644
--- a/mace/core/net.cc
+++ b/mace/core/net.cc
@@ -114,9 +114,8 @@ MaceStatus SerialNet::Run(RunMetadata *run_metadata) {
       }
 
       std::vector<std::vector<int64_t>> output_shapes;
-      for (auto output_shape : op->debug_def().output_shape()) {
-        output_shapes.push_back({output_shape.dims().begin(),
-                                 output_shape.dims().end()});
+      for (auto output : op->Outputs()) {
+        output_shapes.push_back(output->shape());
       }
       OperatorStats op_stats = {op->debug_def().name(), op->debug_def().type(),
                                 output_shapes,