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

Fix segmentation fault caused by invalid tensor type (#4467)

Browse source 
* Fix segmentation fault caused by invalid tensor type

* Addressed review comment
This commit is contained in:
Prabhat 2020-07-10 11:23:12 +01:00 committed by GitHub
parent ccbf49e59f
commit 04586fc09d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/graph/graph.cc
View file

@ -842,6 +842,10 @@ Graph::Graph(const Model& owning_model,
NodeArg* matching_graph_input = GetNodeArg(tensor.name());
TypeProto t{TypeProtoFromTensorProto(tensor)};
if (!utils::HasElemType(t.tensor_type())) {
ORT_THROW("This is an invalid model. Tensor does not have type information.");
}
if (ir_version_ < 4) {
// initializers can have matching graph inputs but are treated as constant,
// so we prefer the shape from the initializer