Update quantization to work with first-class modules. (#21660)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21660
ghimport-source-id: f9a11b2748f49042ee636755358d79c547aa249e
Reviewed By: suo
Differential Revision: D15770237
Pulled By: zdevito
fbshipit-source-id: 41fa8577028eef247bc545635cd93192a0b19db4