diff --git a/tests/brevitas_ort/common.py b/tests/brevitas_ort/common.py index 08d16e9ff..05b6c7394 100644 --- a/tests/brevitas_ort/common.py +++ b/tests/brevitas_ort/common.py @@ -17,10 +17,10 @@ from brevitas.nn import QuantLinear from brevitas.nn import QuantLSTM from brevitas.nn import TruncAvgPool2d -from brevitas.quant.fixed_point import Int8AccumulatorAwareWeightQuant from brevitas.quant.fixed_point import Int8ActPerTensorFixedPoint from brevitas.quant.fixed_point import Int8WeightPerChannelFixedPoint from brevitas.quant.fixed_point import Int8WeightPerTensorFixedPoint +from brevitas.quant.scaled_int import Int8AccumulatorAwareWeightQuant from brevitas.quant.scaled_int import Int8ActPerTensorFloat from brevitas.quant.scaled_int import Int8WeightPerChannelFloat from brevitas.quant.scaled_int import Int8WeightPerTensorFloat