| 24 | |
| 25 | @dataclass(eq=True, frozen=True) |
| 26 | class QuantizationConfig: |
| 27 | input_activation: Optional[QuantizationSpec] |
| 28 | output_activation: Optional[QuantizationSpec] |
| 29 | weight: Optional[QuantizationSpec] |
| 30 | bias: Optional[QuantizationSpec | Callable] |
| 31 | |
| 32 | |
| 33 | def get_quant_config( |
no outgoing calls