I would argue the implementation depends on your coding style and how you would like to use the model.
E.g. if you want to call the submodules separately in the future or would like to easily inspect the output of the first module, you could use the split approach.
However, usually you would implement the model as once class.