saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-30 23:18:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add initializer for sub-graph. (#269)

Browse source
This commit is contained in:
Ke Zhang 2019-01-02 13:10:33 -08:00 committed by GitHub
parent 8f3a492682
commit eb67eadbc6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/graph/function.cc
View file

@ -117,6 +117,14 @@ FunctionImpl::FunctionImpl(const onnxruntime::Graph& graph,
}
sub_graph.AddNode(node->Name(), node->OpType(), node->Description(), inputs, outputs, &node->GetAttributes(), node->Domain());
}
for (auto input : meta_def->inputs) {
const onnx::TensorProto* initializer = nullptr;
if (graph.GetInitializedTensor(input, initializer)) {
sub_graph.AddInitializedTensor(*initializer);
}
}
//TODO: if we reuse the nodes in parent graph, maybe we don't need to resolve it.
ORT_ENFORCE(sub_graph.Resolve().IsOK());
}