Skip to content
Snippets Groups Projects
Commit 01a62e7f authored by auphelia's avatar auphelia
Browse files

[EltWise] Add node attribute for inFIFODepths

parent eabae836
No related branches found
No related tags found
No related merge requests found
...@@ -55,6 +55,7 @@ class StreamingEltwise(HLSCustomOp): ...@@ -55,6 +55,7 @@ class StreamingEltwise(HLSCustomOp):
# [4] is four vectors (like a FC layer with batch=4) # [4] is four vectors (like a FC layer with batch=4)
# [1, 4, 4] is four * four vectors (like a conv layer with batch=1) # [1, 4, 4] is four * four vectors (like a conv layer with batch=1)
"numInputVectors": ("ints", False, [1]), "numInputVectors": ("ints", False, [1]),
"inFIFODepths": ("ints", False, [2, 2]),
} }
my_attrs.update(super().get_nodeattr_types()) my_attrs.update(super().get_nodeattr_types())
return my_attrs return my_attrs
......
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