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):
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),
)
......
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