From 4f2042acd101228d8aded4d05e289a8e16a929a5 Mon Sep 17 00:00:00 2001
From: yao_yf <yaoyifan1@huawei.com>
Date: Mon, 19 Oct 2020 20:19:19 +0800
Subject: [PATCH] move reshape_tensor_info_vector reset position

---
 mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc b/mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc
index 668b817b8d..0c50b847a4 100644
--- a/mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc
+++ b/mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc
@@ -423,7 +423,6 @@ void ReshapeInfo::SetCostForReshape(const mindspore::parallel::StrategyPtr &stra
     std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_);
   swc->cost_list.push_back(result);
   strategy_cost_.emplace_back(swc);
-  ResetQueueMember();
 }
 
 Status ReshapeInfo::GenerateStrategies(int32_t stage_id) {
@@ -489,6 +488,7 @@ Status ReshapeInfo::GenetateStrategyCosts(const std::vector<std::shared_ptr<Stra
       }
       TensorInfo next_in_tensor_info = next_in_tensor_infos[in_index];
       SetOutputLayout(next_in_tensor_info.tensor_layout());
+      ResetQueueMember();
       InferTensorInfoByLayout();
       SetCostForReshape(reshape_stra);
     }