NNgen / nngen

NNgen: A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Incorrect shape in batched processing

shtaxxx opened this issue · comments

When a batch size is greater than 1, the output shape will be incorrect. The cause will be _lazy_reshape inserted between conv2d and matmul.

Fixed in 5dda656