Browse Source

Fix root feed inner Enter

tags/v1.3.0
zhangxiaokun 4 years ago
parent
commit
a35dcece08
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/model/hybrid_model_builder.cc

+ 1
- 1
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -2358,7 +2358,7 @@ Status HybridModelBuilder::BuildFrameGroupIndex(NodeItem &node_item) {
int64_t ctrl_flow_group = -1;
if (node_item.IsEnterOp() && AttrUtils::GetInt(node_item.op_desc, ATTR_NAME_CONTROL_FLOW_GROUP, ctrl_flow_group)) {
node_item.frame_index_ = ctrl_flow_group;
for (const auto src_node : node_item.node->GetInDataNodes()) {
for (const auto src_node : node_item.node->GetInAllNodes()) {
NodeItem *src_node_item = nullptr;
GE_CHK_STATUS_RET(GetOrCreateNodeItem(src_node, &src_node_item),
"[%s] failed to get or create node item", src_node->GetName().c_str());


Loading…
Cancel
Save