From cde63ec14878c29cfd87867463e31fe2dbfd4178 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 14 Oct 2021 16:50:21 +0200
Subject: [PATCH] [Lookup] minor fix to shape inference
---
src/finn/custom_op/fpgadataflow/lookup.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/lookup.py b/src/finn/custom_op/fpgadataflow/lookup.py
index e7a957169..6a7848ccd 100644
--- a/src/finn/custom_op/fpgadataflow/lookup.py
+++ b/src/finn/custom_op/fpgadataflow/lookup.py
@@ -79,10 +79,10 @@ class Lookup(HLSCustomOp):
return self.get_normal_output_shape()
def make_shape_compatible_op(self, model):
- exp_ishape = self.get_normal_input_shape()
- oshape = self.get_normal_output_shape()
+ exp_ishape = tuple(self.get_normal_input_shape())
+ oshape = tuple(self.get_normal_output_shape())
ishape = tuple(model.get_tensor_shape(self.onnx_node.input[0]))
- assert ishape == exp_ishape, "Unexpect input shape for Lookup: %s vs %s" % (
+ assert ishape == exp_ishape, "Unexpected input shape for Lookup: %s vs %s" % (
str(exp_ishape),
str(ishape),
)
--
GitLab