Is your model working generally without splitting it?
The error points towards a shape mismatch in a conv layer. If the model is working in the original form, make sure to apply all functional API calls, such as flattening the activation tensors etc.