Removing input_planes_dim in convolutions