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 …]
|