Huawei_Technology
/
MindSpore-GraphEngine
Not watched
1
0
Fork 0
Code
Releases 10 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

For dynamic shape compile.

pull/693/head
unknown 5 years ago
parent
84ecdd25f5
commit
b86a5ce46d
1 changed files with 1 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      ge/graph/build/graph_builder.cc

+ 1
- 0
ge/graph/build/graph_builder.cc View File

@@ -312,6 +312,7 @@ Status GraphBuilder::SetConstantInputOffset(ComputeGraphPtr &comp_graph) {
std::vector<int64_t> output_offsets(op_desc->GetOutputsSize(), 0); std::vector<int64_t> output_offsets(op_desc->GetOutputsSize(), 0);
op_desc->SetOutputOffset(output_offsets); op_desc->SetOutputOffset(output_offsets);
} }
return SUCCESS;
} }


Status GraphBuilder::BuildForUnknownShapeGraph(ComputeGraphPtr &comp_graph, GeModelPtr &ge_model_ptr, Status GraphBuilder::BuildForUnknownShapeGraph(ComputeGraphPtr &comp_graph, GeModelPtr &ge_model_ptr,


Write Preview
Loading…
Cancel
Save