Home
last modified time | relevance | path

Searched refs:eager_quantizable_model (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py9281 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
9494eager_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 …]