Method register_network_output

tensorrt_llm/module.py:192–193

(self, name, value)

            self._parameters[name] = param

    def register_network_output(self, name, value):
        self._network_outputs[name] = value

    def named_network_outputs(self):
        for name, module in self.named_modules():