with_cpu
Fix memroy acces in batchnorm, update resnet training script
Add tests for Linear layer and update matmul usage