From 644b680df3c82cbbe5a13bcb654a39dc7e9df297 Mon Sep 17 00:00:00 2001
From: Ziyan <gongziyan1@huawei.com>
Date: Tue, 8 Sep 2020 20:21:32 +0800
Subject: [PATCH] fix parallel optimizer bug and add st testcase

---
 .../ccsrc/frontend/operator/ops_front_infer_function.cc    | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/frontend/operator/ops_front_infer_function.cc b/mindspore/ccsrc/frontend/operator/ops_front_infer_function.cc
index 68912c0b40..b6e679ee79 100644
--- a/mindspore/ccsrc/frontend/operator/ops_front_infer_function.cc
+++ b/mindspore/ccsrc/frontend/operator/ops_front_infer_function.cc
@@ -636,7 +636,12 @@ AbstractBasePtr InferImplAssign(const AnalysisEnginePtr &, const PrimitivePtr &p
   CheckArgsSize(primitive->name(), args_spec_list, 2);
 
   MS_LOG(DEBUG) << "InferImplAssign " << args_spec_list[0];
-  return args_spec_list[0];
+  auto type = args_spec_list[0]->BuildType();
+  if (type->type_id() == kObjectTypeRefKey) {
+    return args_spec_list[1]->Broaden();
+  } else {
+    return args_spec_list[0];
+  }
 }
 
 REGISTER_FRONTENT_PRIMITIVE_EVAL_IMPL(TypeOf, prim::kPrimTypeOf, InferImplTypeof);