From 02abad9a8fd8f96728a93cf95d5ce970d491e793 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 7 Feb 2020 20:20:18 +0100
Subject: [PATCH] [Driver] use tuples for shapes
---
src/finn/transformation/fpgadataflow/make_pynq_driver.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/transformation/fpgadataflow/make_pynq_driver.py b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
index a35e70591..a84518872 100644
--- a/src/finn/transformation/fpgadataflow/make_pynq_driver.py
+++ b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
@@ -35,8 +35,8 @@ class MakePYNQDriver(Transformation):
# TODO convert this to an analysis pass
i_tensor_name = model.graph.input[0].name
o_tensor_name = model.graph.output[0].name
- i_tensor_shape = model.get_tensor_shape(i_tensor_name)
- o_tensor_shape = model.get_tensor_shape(o_tensor_name)
+ i_tensor_shape = tuple(model.get_tensor_shape(i_tensor_name))
+ o_tensor_shape = tuple(model.get_tensor_shape(o_tensor_name))
i_tensor_dt = model.get_tensor_datatype(i_tensor_name)
o_tensor_dt = model.get_tensor_datatype(o_tensor_name)
# generate dummy i/o tensors and their packed versions
--
GitLab