From 163f2577006f0a32f38fbabe9b39752a36e25c19 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Fri, 1 Nov 2019 16:54:30 +0000
Subject: [PATCH] [Execution] Committed current (not working!) version of the
shape inference
---
src/finn/transformation/infer_shapes.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/finn/transformation/infer_shapes.py b/src/finn/transformation/infer_shapes.py
index bc8b5ccf5..d834428dd 100644
--- a/src/finn/transformation/infer_shapes.py
+++ b/src/finn/transformation/infer_shapes.py
@@ -5,6 +5,9 @@ def infer_shapes(model):
"""Ensure every tensor in the model has a specified shape (ValueInfo)."""
for node in model.graph.node:
+ for output in node.output:
+ model.set_tensor_shape(output, [1])
+
# create an empty execution context
execution_context = model.make_empty_exec_context()
--
GitLab