Update eq_generator to support multi inputs