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 7810df6 commit e691e92Copy full SHA for e691e92
py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py
@@ -416,3 +416,5 @@ def output(self, target, args, kwargs):
416
elif self.output_fp16 and output.dtype == trt.float32:
417
output.dtype = trt.float16
418
self._output_names.append(name)
419
+
420
+ return list(outputs)
0 commit comments