Browse Source

update

pull/1317/head
chuxing 4 years ago
parent
commit
cd6ca2071b
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      ge/hybrid/model/hybrid_model_builder.cc

+ 5
- 0
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -1077,7 +1077,12 @@ Status HybridModelBuilder::InitWeights() {

Status HybridModelBuilder::LoadTasks() {
GE_CHK_STATUS_RET(CheckAicpuOpList(), "Check Aicpu op failed.");
std::map<int64_t, NodeItem *> ordered_node_items;
for (auto &it : hybrid_model_.node_items_) {
auto &node_item = it.second;
ordered_node_items.emplace(node_item->node_id, node_item.get());
}
for (auto &it : ordered_node_items) {
auto &node_item = it.second;
auto &node_ptr = node_item->node;
if (node_item->node_type == NETOUTPUT) {


Loading…
Cancel
Save