Method get_input

tensorrt_llm/graph_rewriting.py:366–367

(self, name: str)

        self.raw_outputs = outputs

    def get_input(self, name: str) -> Any:
        return self.raw_inputs[name]

    def clone_inputs(self):
        """Get a shallow copy of the inputs."""