Searched refs:quantization_capture (Results 1 – 2 of 2) sorted by relevance
17 def __init__(self, quantization_capture=False): argument24 self.quantization_capture = quantization_capture25 if quantization_capture:40 if self.quantization_capture:
198 model = RecomposePixelUnshuffle(quantization_capture=True)(model).graph_module