[WIP] Fix cifar10 format

Fix bug reported about performance loss on cifar10, traced back to a misconversion from design matrix to conv2d space.
Still needed:
- make sure all tests pass
- new test for that issue