Add support for cross generalization for time_gen