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

[StreamingFC] add outer dummy dimension to thresholds

hlslib threshold init seems to fail otherwise
parent e1f2d337
No related branches found
No related tags found
No related merge requests found
......@@ -146,7 +146,7 @@ class StreamingFCLayer_Batch(HLSCustomOp):
assert ret.shape[0] == pe
assert ret.shape[1] == tmem
assert ret.shape[2] == n_thres_steps
return ret
return ret.reshape(1, pe, tmem, n_thres_steps)
def generate_params(self, model):
# weights
......
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