Incorrect shape in batched processing

NNgen / nngen

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

Apache License 2.0

340 stars 46 forks source link

Closed shtaxxx closed 4 years ago

shtaxxx commented 4 years ago

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.

shtaxxx commented 4 years ago

Fixed in 5dda656be4ab8764c0139c347e532b83663d40b5