Method post_forward_hook

ppq/quantization/analyse/graphwise.py:53–57

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

        return super().pre_forward_hook(inputs, **kwargs)

    def post_forward_hook(self, outputs: List[torch.Tensor], **kwargs) -> list:
        for idx, output in enumerate(outputs):
            self.o_storage[idx].append(tensor_random_fetch(
            output, seed=10086, num_of_fetches=self.fetchs).to('cpu'))
        return super().post_forward_hook(outputs, **kwargs)

    def clear(self):
        self.i_storage = [[] for _ in range(self._hook_to.num_of_input)]