We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 65277c5 commit 95730feCopy full SHA for 95730fe
py/torch_tensorrt/dynamo/conversion/trt_interpreter.py
@@ -356,3 +356,5 @@ def output(self, target, args, kwargs):
356
elif self.output_fp16 and output.dtype == trt.float32:
357
output.dtype = trt.float16
358
self._output_names.append(name)
359
+
360
+ return list(outputs)
0 commit comments