Augmented Generators for Non-autonomous Flows