Method tracing_operation_meta

ppq/executor/torch.py:581–602

Tracing meta data for each operation, if there are some already created meta data with your operation, They will be override without warrning. Args: inputs (Union[dict, list, torch.Tensor]): [description] output_names (List[str], optional): [descripti

(
        self,
        inputs: Union[dict, list, torch.Tensor],
        output_names: List[str] = None,
    )

    @ torch.no_grad()
    @ empty_ppq_cache
    def tracing_operation_meta(
        self,
        inputs: Union[dict, list, torch.Tensor],
        output_names: List[str] = None,
    ) -> None:
        """Tracing meta data for each operation, if there are some already
        created meta data with your operation, They will be override without
        warrning.

        Args:
            inputs (Union[dict, list, torch.Tensor]): [description]
            output_names (List[str], optional): [description]. Defaults to None.
        """
        hooks = {}
        for op_name, operation in self._graph.operations.items():
            hooks[op_name] = TorchMetaDataTracingHook(operation=operation)

        self.__forward(
            inputs=inputs,
            output_names=output_names,
            executing_order=self._executing_order,
            hooks=hooks)

    def load_graph(self, graph: BaseGraph) -> dict:
        super().load_graph(graph)