From 9efcf9ee6948ffb9e54d69992ed5dca879869fab Mon Sep 17 00:00:00 2001
From: xuanyue <xuanyue@huawei.com>
Date: Thu, 28 Jan 2021 10:36:45 +0800
Subject: [PATCH] fix resize for mindir and adjust onnx dropout

---
 .../lite/tools/converter/parser/onnx/onnx_dropout_parser.cc    | 2 +-
 mindspore/lite/tools/optimizer/graph/primitive_adjust_pass.cc  | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.cc b/mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.cc
index 418eaf3197..195756c4a1 100644
--- a/mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.cc
+++ b/mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.cc
@@ -30,7 +30,7 @@ ops::PrimitiveC *OnnxDropoutParser::Parse(const onnx::GraphProto &onnx_graph, co
   for (const auto &onnx_node_attr : onnx_node.attribute()) {
     const auto &attribute_name = onnx_node_attr.name();
     if (attribute_name == "ratio") {
-      primitive_c->set_ratio(onnx_node_attr.f());
+      primitive_c->set_keep_prob(onnx_node_attr.f());
     }
   }
 
diff --git a/mindspore/lite/tools/optimizer/graph/primitive_adjust_pass.cc b/mindspore/lite/tools/optimizer/graph/primitive_adjust_pass.cc
index ff4651d113..c08427119b 100644
--- a/mindspore/lite/tools/optimizer/graph/primitive_adjust_pass.cc
+++ b/mindspore/lite/tools/optimizer/graph/primitive_adjust_pass.cc
@@ -401,10 +401,11 @@ int MoveAttrMapResize(const CNodePtr &cnode) {
     return lite::RET_ERROR;
   }
   auto dst_prim = std::make_shared<ops::Resize>();
+  MS_ASSERT(dst_prim != nullptr);
   auto size = GetValue<std::vector<int64_t>>(src_prim->GetAttr(ops::kSize));
   dst_prim->set_new_height(size[0]);
   dst_prim->set_new_width(size[1]);
-  if (dst_prim->GetAttr(ops::kAlignCorners) != nullptr && GetValue<bool>(dst_prim->GetAttr(ops::kAlignCorners))) {
+  if (src_prim->GetAttr(ops::kAlignCorners) != nullptr && GetValue<bool>(src_prim->GetAttr(ops::kAlignCorners))) {
     dst_prim->set_coordinate_transform_mode(mindspore::ALIGN_CORNERS);
   }
   if (src_prim->name() == kNameResizeBilinear) {