batch train for bert 2.x

parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc

@@ -109,6 +109,9 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
         return FAILED;
       }
     }
+    const auto out_desc = op_dest->MutableOutputDesc(0);
+    GE_CHECK_NOTNULL(out_desc);
+    out_desc->SetDataType(out_type);
 
     std::shared_ptr<NodeDef> pkg_node = ge::parser::MakeShared<NodeDef>();
     GE_CHECK_NOTNULL(pkg_node);

tests/st/testcase/test_tensorflow_parser.cc

@@ -1913,6 +1913,7 @@ TEST_F(STestTensorflowParser, tensorflow_auto_mapping_parser_adapter_test)
   EXPECT_EQ(ret, SUCCESS);
 
   op_dest->SetType(ge::parser::SHAPE);
+  op_dest->AddOutputDesc(GeTensorDesc());
   ret = autoMappingParser.ParseParams(node_def, op_dest);
   EXPECT_EQ(ret, SUCCESS);
 }

tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc

@@ -2082,6 +2082,7 @@ TEST_F(UtestTensorflowParser, tensorflow_auto_mapping_parser_adapter_test)
   EXPECT_EQ(ret, SUCCESS);
 
   op_dest->SetType(ge::parser::SHAPE);
+  op_dest->AddOutputDesc(GeTensorDesc());
   ret = autoMappingParser.ParseParams(node_def, op_dest);
   EXPECT_EQ(ret, SUCCESS);
 }