Method get_outputs

python/tvm/runtime/vm.py:294–323

Get the value output by the function by the given name after a call of `invoke_stateful`. It is an error to call this function without first calling `invoke_stateful`. Parameters ---------- func_name: str The name of the function whose o

(self, func_name: str)

        self._invoke_stateful(func_name)

    def get_outputs(self, func_name: str) -> tvm.Object | tuple[Any]:
        """
        Get the value output by the function by the given name
        after a call of `invoke_stateful`.

        It is an error to call this function without first calling `invoke_stateful`.

        Parameters
        ----------
        func_name: str
            The name of the function whose output should be fetched.

        Returns
        -------
        ret: Union[tvm.Object, Tuple[Any]]
            The result of the earlier call to the function via `invoke_stateful`.
            If the result is a tuple, it returns a list of the fields.
            The fields are potentially also tuples, so these can be arbitrily nested.
        """

        # to deal with potentially nested tuples, we need to query for arity recursively
        def get_output_rec(func_name, *idx):
            arity = self._get_output_arity(func_name, *idx)
            if arity == -1:
                return self._get_output(func_name, *idx)
            # otherwise we need to specify more indices
            idx_list = list(idx)
            return tuple(get_output_rec(func_name, *(idx_list + [i])) for i in range(arity))

        return get_output_rec(func_name)

    def set_instrument(self, instrument: Function) -> None:
        """Set an instrumentation function.