input_generator: Add a new API for initialize