with_cpu
Variable constructor refactoring + ReLU+ LeakyReLU
Batchnorm2d backward implementation
Conv2d backward pass implementation
Dispatcher implementation
Fix dispatcher calls and add log_softmax test
Refactor Shape class
MaxPool2d implementation
Init functions and checks
Batchnorm forward tests