Method post_forward_hook

ppq/executor/torch.py:34–40

(self, outputs: List[torch.Tensor], **kwargs)

        return inputs

    def post_forward_hook(self, outputs: List[torch.Tensor], **kwargs) -> list:
        for tensor, var in zip(outputs, self._hook_to.outputs):
            if tensor is not None:
                var.shape = tensor.shape
                var.dtype = tensor.dtype

        return outputs


class TorchQuantizeDelegator(Callable):