Searched refs:eager_quantizable_model (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_quantize_fx.py | 9281 self, mode, name, model, eager_quantizable_model, argument 9352 if eager_quantizable_model is not None: 9354 qeager = eager_quantizable_model 9492 eager_quantizable_model = None 9494 …eager_quantizable_model = quantized_models.__dict__[name](pretrained=False, quantize=False, **kwar… 9496 pretrained = eager_quantizable_model is not None 9504 if eager_quantizable_model: 9505 eager_quantizable_model.aux_logits = False 9506 eager_quantizable_model.AuxLogits = None 9510 mode, name, model, eager_quantizable_model, [all …]
|