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

[Lookup] minor fix to shape inference

parent 90e1cf43
No related branches found
No related tags found
No related merge requests found
...@@ -79,10 +79,10 @@ class Lookup(HLSCustomOp): ...@@ -79,10 +79,10 @@ class Lookup(HLSCustomOp):
return self.get_normal_output_shape() return self.get_normal_output_shape()
def make_shape_compatible_op(self, model): def make_shape_compatible_op(self, model):
exp_ishape = self.get_normal_input_shape() exp_ishape = tuple(self.get_normal_input_shape())
oshape = self.get_normal_output_shape() oshape = tuple(self.get_normal_output_shape())
ishape = tuple(model.get_tensor_shape(self.onnx_node.input[0])) 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(exp_ishape),
str(ishape), str(ishape),
) )
......
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