Browse Source

update parameter

pull/616/head
taoxiangdong 5 years ago
parent
commit
80795addb1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/model_builder.cc

+ 1
- 1
ge/graph/build/model_builder.cc View File

@@ -609,7 +609,7 @@ Status ModelBuilder::SaveDataToModel(ge::Model &model, ge::GeModel &ge_model) {
tbe_kernel_store_.AddTBEKernel(tbe_kernel); tbe_kernel_store_.AddTBEKernel(tbe_kernel);
} }


(void)SetModelCheckAicpuAttr(model);
(void)SetModelCheckAicpuAttr(model, aicpu_optype_set, aicpu_tf_optype_set);


for (const ge::NodePtr &n : compute_graph_->GetNodes(compute_graph_->GetGraphUnknownFlag())) { for (const ge::NodePtr &n : compute_graph_->GetNodes(compute_graph_->GetGraphUnknownFlag())) {
auto node_op_desc = n->GetOpDesc(); auto node_op_desc = n->GetOpDesc();


Loading…
Cancel
Save