Skip to content
Snippets Groups Projects
Commit 1cb6fad9 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Exec] add check for all shapes specified

parent cf4472fd
No related branches found
No related tags found
No related merge requests found
......@@ -67,6 +67,9 @@ def execute_onnx(model, input_dict, return_full_exec_context=False):
the execution (including inputs, weights, activations and final outputs)
will be returned as a dict."""
if not model.check_all_tensor_shapes_specified():
raise Exception("Found unspecified tensor shapes, try infer_shapes")
graph = model.graph
# first, we need to make sure that every variable required by the graph has
# some buffer associated with it. this includes graph inputs (which includes
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment