Method set_runtime_tensor_shape

scripts/deployment/trt_torch.py:131–133

Set runtime input shape for dynamic dimensions.

(self, name, shape)

        raise KeyError(f"Input tensor '{tensor_name}' not found in engine.")

    def set_runtime_tensor_shape(self, name, shape):
        """Set runtime input shape for dynamic dimensions."""
        self.execution_context.set_input_shape(name, shape)

    def forward(self, *args, **kwargs):
        """Execute TRT inference with the given input tensors.