Keeping DP override in sync with upstream torch
Same as in #4630, we want to redirect forward to the training_step, validaion_step, test_step methods but without having to copy the DP implementation directly from pytorch codebase. This would also allow users to easily override DP in Lightning.
See refactor for DDP here: #5185