Rename need_infer_again_

pull/1318/head
zhangxiaokun 4 years ago
parent 000b53aeb5
commit 7fbfe1467f

@ -43,7 +43,7 @@ Status InferShapePass::Run(NodePtr &node) {
return GE_GRAPH_INFERSHAPE_FAILED;
}
bool need_repass = false;
auto has_attr = AttrUtils::GetBool(node->GetOpDesc(), "need_infer_again_", need_repass);
auto has_attr = AttrUtils::GetBool(node->GetOpDesc(), "_need_infer_again", need_repass);
if (has_attr) {
if (!OptionExists(kOptimizeAfterSubGraph)) {
return SUCCESS;
@ -53,7 +53,7 @@ Status InferShapePass::Run(NodePtr &node) {
GELOGD("Node %s need repass immediately.", node->GetName().c_str());
} else {
// clear attr on while
node->GetOpDesc()->DelAttr("need_infer_again_");
node->GetOpDesc()->DelAttr("_need_infer_again");
}
}
return SUCCESS;

@ -70,17 +70,17 @@ class UtestTestPass : public BaseNodePass {
// simulate infershape pass
if(node->GetType() == WHILE){
bool need_repass = false;
AttrUtils::GetBool(node->GetOpDesc(),"need_infer_again_", need_repass);
AttrUtils::GetBool(node->GetOpDesc(),"_need_infer_again", need_repass);
if(!OptionExists(kOptimizeAfterSubGraph)){
return SUCCESS;
}
if(need_repass){
AttrUtils::SetBool(node->GetOpDesc(),"need_infer_again_", false);
AttrUtils::SetBool(node->GetOpDesc(),"_need_infer_again", false);
AddImmediateRePassNode(node);
}
else{
// clear attr on while
node->GetOpDesc()->DelAttr("need_infer_again_");
node->GetOpDesc()->DelAttr("_need_infer_again");
}
}
return SUCCESS;
@ -492,7 +492,7 @@ ComputeGraphPtr BuildWhileGraph1() {
for (int i = 0; i < 2; ++i) {
op_desc->AddOutputDesc(tensor_desc->Clone());
}
AttrUtils::SetBool(op_desc,"need_infer_again_", true);
AttrUtils::SetBool(op_desc,"_need_infer_again", true);
op_desc->AddSubgraphName(sub_graph->GetName());
op_desc->SetSubgraphInstanceName(0,sub_graph->GetName());
auto root_graph = builder.GetGraph();

Loading…
Cancel
Save