Browse Source

dynamic shape inference support

pull/1240/head
lichun 5 years ago
parent
commit
9ac8c3e181
1 changed files with 7 additions and 5 deletions
  1. +7
    -5
      ge/hybrid/model/node_item.cc

+ 7
- 5
ge/hybrid/model/node_item.cc View File

@@ -152,11 +152,13 @@ Status NodeItem::InitInputsAndOutputs() {
if (AttrUtils::GetInt(node->GetOwnerComputeGraph(), ::ge::ATTR_STAGE_LEVEL, group)) {
GELOGD("[%s] Got stage level from parent graph = %d", op_desc->GetName().c_str(), group);
} else {
auto parent_node = node->GetOwnerComputeGraph()->GetParentNode();
if ((parent_node != nullptr) && (AttrUtils::GetInt(parent_node->GetOpDesc(), ::ge::ATTR_STAGE_LEVEL, group))) {
GELOGD("[%s] Got stage level from parent node = %d", op_desc->GetName().c_str(), group);
} else {
GELOGD("[%s] Node do not set stage level", op_desc->GetName().c_str());
if (node->GetOwnerComputeGraph() != nullptr) {
auto parent_node = node->GetOwnerComputeGraph()->GetParentNode();
if ((parent_node != nullptr) && (AttrUtils::GetInt(parent_node->GetOpDesc(), ::ge::ATTR_STAGE_LEVEL, group))) {
GELOGD("[%s] Got stage level from parent node = %d", op_desc->GetName().c_str(), group);
} else {
GELOGD("[%s] Node do not set stage level", op_desc->GetName().c_str());
}
}
}
}


Loading…
Cancel
Save