RepQ: Generalizing Quantization-Aware Training for Re-Parametrized Architectures