From 1cb6fad92d5cb3015e79ff0e256b2f0e9d375ff2 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 22 Oct 2019 14:58:51 +0100
Subject: [PATCH] [Exec] add check for all shapes specified
---
src/finn/core/onnx_exec.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/finn/core/onnx_exec.py b/src/finn/core/onnx_exec.py
index 30cbcc6a3..5521bccbd 100644
--- a/src/finn/core/onnx_exec.py
+++ b/src/finn/core/onnx_exec.py
@@ -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
--
GitLab