aten::cat converter moving to impl

Author: apbose

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

@@ -60,7 +60,7 @@ def aten_ops_cat(
         SourceIR.ATEN,
         name,
         tensors=args[0],
-        dim = args_bounds_check(args, 2, 1),
+        dim=args_bounds_check(args, 2, 1),
     )
 
 

py/torch_tensorrt/dynamo/conversion/impl/cat.py

@@ -1,21 +1,20 @@
-from typing import Optional, Union, Sequence, Dict
+from typing import Dict, Optional, Sequence, Union
 
 import torch
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
 from torch_tensorrt.fx.converters.converter_utils import set_layer_name
 from torch_tensorrt.fx.types import TRTNetwork, TRTTensor
 
+
 def cat(
     network: TRTNetwork,
     target: Target,
     source_ir: Optional[SourceIR],
     name: str,
     input: TRTNetwork,
     dim: int,
-    
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-
     if any(not isinstance(t, TRTTensor) for t in input):  # type: ignore[union-attr]
         raise RuntimeError(
             f"cat received inputs {input} that is not part " "of the TensorRT region!"
@@ -26,4 +25,4 @@ def cat(
 
     concat_layer.axis = dim
     set_layer_name(concat_layer, target, name + "_gather", source_ir)
-    return concat_layer.get_output(0)
+    return concat_layer.get_output(0)