Temporary adding a name for PerTensorQuantizer for backwards compatib…

…ility
ethanhe42 · Jun 17, 2021 · e5284d6 · e5284d6
1 parent a368882
commit e5284d6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/tensor_quantizer.py b/TrainingExtensions/torch/src/python/aimet_torch/tensor_quantizer.py
@@ -229,6 +229,7 @@ def set_encoding(self, encoding: libpymo.TfEncoding):
 
 # Temporary change to preserve backwards compatibility
 PostTrainingTensorQuantizer = StaticGridTensorQuantizer
+StaticGridPerTensorQuantizer = StaticGridTensorQuantizer
 
 
 class QuantizeDequantize(torch.autograd.Function):