Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A D_learnable_fake_quantize.py95 self.toggle_qparam_learning(enabled=True).toggle_fake_quant(
107 self.toggle_qparam_learning(enabled=False).toggle_fake_quant(
118 self.toggle_qparam_learning(enabled=False).toggle_fake_quant(
132 def toggle_qparam_learning(self, enabled=True): member in _LearnableFakeQuantize