Changed the error type when Conv2D given wrong arguments