(self, neutron_target_spec: NeutronTargetSpec, is_qat: bool = False)
| 236 | |
| 237 | class NeutronQuantizer(ComposableQuantizer): |
| 238 | def __init__(self, neutron_target_spec: NeutronTargetSpec, is_qat: bool = False): |
| 239 | self.neutron_target_spec = neutron_target_spec |
| 240 | self.is_qat = is_qat |
| 241 | |
| 242 | static_qconfig = QuantizationConfig( |
| 243 | act_qspec(is_qat=is_qat), |
| 244 | act_qspec(is_qat=is_qat), |
| 245 | wgt_qspec(is_qat=is_qat), |
| 246 | None, |
| 247 | ) |
| 248 | static_fc_qconfig = QuantizationConfig( |
| 249 | act_qspec(is_qat=is_qat), |
| 250 | act_qspec(is_qat=is_qat), |
| 251 | wgt_fc_qspec(is_qat=is_qat), |
| 252 | None, |
| 253 | ) |
| 254 | |
| 255 | OpQuantizer = NeutronAtenQuantizer |
| 256 | super().__init__( |
| 257 | [ |
| 258 | OpQuantizer(AbsPattern(is_qat=is_qat), static_qconfig), |
| 259 | OpQuantizer(AdaptiveAvgPoolPattern(is_qat=is_qat), static_qconfig), |
| 260 | OpQuantizer(AddTensorPattern(is_qat=is_qat), static_qconfig), |
| 261 | OpQuantizer(AddmmPattern(self, is_qat=is_qat), static_fc_qconfig), |
| 262 | OpQuantizer(AvgPool1DPattern(is_qat=is_qat), static_qconfig), |
| 263 | OpQuantizer(AvgPool2DPattern(is_qat=is_qat), static_qconfig), |
| 264 | OpQuantizer(BatchNormPattern(is_qat=is_qat), static_qconfig), |
| 265 | OpQuantizer(BMMPattern(is_qat=is_qat), static_qconfig), |
| 266 | OpQuantizer(CatPattern(is_qat=is_qat), static_qconfig), |
| 267 | OpQuantizer(ClampPattern(is_qat=is_qat), static_qconfig), |
| 268 | OpQuantizer(Conv2dPattern(self, is_qat=is_qat), static_qconfig), |
| 269 | OpQuantizer( |
| 270 | ConvTranspose2dPattern(self, is_qat=is_qat), static_qconfig |
| 271 | ), |
| 272 | OpQuantizer(DropoutPattern(is_qat=is_qat), static_qconfig), |
| 273 | OpQuantizer(FlattenPattern(is_qat=is_qat), static_qconfig), |
| 274 | OpQuantizer(HardTanhPattern(is_qat=is_qat), static_qconfig), |
| 275 | OpQuantizer(HardTanhInPlacePattern(is_qat=is_qat), static_qconfig), |
| 276 | OpQuantizer(LeakyReluPattern(is_qat=is_qat), static_fc_qconfig), |
| 277 | OpQuantizer(LeakyReluInPlacePattern(is_qat=is_qat), static_fc_qconfig), |
| 278 | OpQuantizer(LinearPattern(self, is_qat=is_qat), static_fc_qconfig), |
| 279 | OpQuantizer(MaxPool1DPattern(is_qat=is_qat), static_qconfig), |
| 280 | OpQuantizer(MaxPool2DPattern(is_qat=is_qat), static_qconfig), |
| 281 | OpQuantizer(MeanDimPattern(is_qat=is_qat), static_qconfig), |
| 282 | OpQuantizer(MmPattern(self, is_qat=is_qat), static_qconfig), |
| 283 | OpQuantizer(MulTensorPattern(is_qat=is_qat), static_qconfig), |
| 284 | OpQuantizer(NegPattern(is_qat=is_qat), static_qconfig), |
| 285 | OpQuantizer(PadPattern(is_qat=is_qat), static_qconfig), |
| 286 | OpQuantizer(PermutePattern(is_qat=is_qat), static_qconfig), |
| 287 | OpQuantizer(PReLUPattern(is_qat=is_qat), static_qconfig), |
| 288 | OpQuantizer(ReluPattern(is_qat=is_qat), static_qconfig), |
| 289 | OpQuantizer(ReluInPlacePattern(is_qat=is_qat), static_qconfig), |
| 290 | OpQuantizer(ReshapePattern(is_qat=is_qat), static_qconfig), |
| 291 | OpQuantizer(SigmoidPattern(is_qat=is_qat), static_qconfig), |
| 292 | OpQuantizer(SliceTensorPattern(is_qat=is_qat), static_qconfig), |
| 293 | OpQuantizer(SoftMaxPattern(is_qat=is_qat), static_qconfig), |
| 294 | OpQuantizer(SqueezeDimPattern(is_qat=is_qat), static_qconfig), |
| 295 | OpQuantizer(SqueezeDimsPattern(is_qat=is_qat), static_qconfig), |
no test coverage detected