Is it possible to convert a torch traced model that accepts List[torch.Tensor] type of input to trt ts module?