Method __call__

ppq/samples/custimize_quant_func.py:18–22

(self, tensor: torch.Tensor, config: TensorQuantizationConfig)

    register your delegator with executor.register_quantize_delegate
    """
    def __call__(self, tensor: torch.Tensor, config: TensorQuantizationConfig) -> torch.Tensor:
        if config.policy.has_property(QuantizationProperty.ASYMMETRICAL):
            raise ValueError('Sorry, this delegator handles only Symmetrical Quantizations.')
        print('You are invoking cusitmized quant function now.')
        return torch.round(tensor / config.scale) * config.scale

BATCHSIZE = 32
INPUT_SHAPE = [3, 224, 224]