| @@ -826,19 +826,20 @@ bool BlockMemAssigner::IsContinuousOutput(const NodePtr &n) { | |||||
| return false; | return false; | ||||
| } | } | ||||
| auto node_desc = n->GetOpDesc(); | |||||
| if (node_desc == nullptr) { | |||||
| GELOGE(FAILED, "Node[%s] nodedesc is null.", n->GetName().c_str()); | |||||
| return false; | |||||
| } | |||||
| // if output size just one, no need to reassign continuous memory | // if output size just one, no need to reassign continuous memory | ||||
| if (node_op_desc->GetOutputsSize() == 1) { | |||||
| if (node_desc->GetOutputsSize() == 1) { | |||||
| GELOGI("op %s output size is one, no need to continuous process.", n->GetName().c_str()); | GELOGI("op %s output size is one, no need to continuous process.", n->GetName().c_str()); | ||||
| return false; | return false; | ||||
| } | } | ||||
| // Get the continuous output type of the node, default is false | // Get the continuous output type of the node, default is false | ||||
| bool is_output_continuous = false; | bool is_output_continuous = false; | ||||
| auto node_desc = n->GetOpDesc(); | |||||
| if (node_desc == nullptr) { | |||||
| GELOGE(FAILED, "Node[%s] nodedesc is null.", n->GetName().c_str()); | |||||
| return false; | |||||
| } | |||||
| // If GetBool fail, is_output_continuous is false. | // If GetBool fail, is_output_continuous is false. | ||||
| (void)ge::AttrUtils::GetBool(node_desc, ATTR_NAME_CONTINUOUS_OUTPUT, is_output_continuous); | (void)ge::AttrUtils::GetBool(node_desc, ATTR_NAME_CONTINUOUS_OUTPUT, is_output_continuous); | ||||