[Quantization] Problems with recent refactoring changes in the quantization pass

[tico]

Hi @vinx13!

thanks for the update! Let us know once this patch has been merged in the master to give it try.