Method set_runtime_tensor_shape

groot/control/tensorrt_utils.py:488–489

(self, name, shape)

        return self.forward(*args, **inputs)

    def set_runtime_tensor_shape(self, name, shape):
        self.execution_context.set_input_shape(name, shape)

    def forward(self, *args, **kwargs):
        return_list = kwargs.pop("return_list", False)