未验证 提交 a623ce04 编写于 作者: C cc 提交者: GitHub

Use different name_scope for different conv type, test=develop (#29355)

上级 8fc7f1b6
...@@ -171,7 +171,7 @@ void ConvBNFusePass::ApplyImpl(ir::Graph* graph) const { ...@@ -171,7 +171,7 @@ void ConvBNFusePass::ApplyImpl(ir::Graph* graph) const {
// Create eltwise_y (conv bias) variable // Create eltwise_y (conv bias) variable
VarDesc eltwise_y_in_desc( VarDesc eltwise_y_in_desc(
patterns::PDNodeName(name_scope_, "eltwise_y_in")); patterns::PDNodeName("fuse_conv_bn", conv_type() + "_eltwise_y_in"));
eltwise_y_in_desc.SetShape(framework::vectorize(bn_bias_tensor->dims())); eltwise_y_in_desc.SetShape(framework::vectorize(bn_bias_tensor->dims()));
eltwise_y_in_desc.SetDataType(bn_bias_tensor->type()); eltwise_y_in_desc.SetDataType(bn_bias_tensor->type());
eltwise_y_in_desc.SetLoDLevel(bn_bias->Var()->GetLoDLevel()); eltwise_y_in_desc.SetLoDLevel(bn_bias->Var()->GetLoDLevel());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册