File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
py/torch_tensorrt/dynamo/conversion/impl Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1616)
1717
1818
19- def unify_trt_tensors (
19+ def unify_and_concat_trt_tensors (
2020 ctx : ConversionContext ,
2121 target : Target ,
2222 name : str ,
@@ -115,7 +115,7 @@ def cat(
115115 trt_promoted_type = None
116116
117117 dim = get_positive_dim (dim , len (trt_inputs [0 ].shape ))
118- return unify_trt_tensors (
118+ return unify_and_concat_trt_tensors (
119119 ctx ,
120120 target ,
121121 name ,
Original file line number Diff line number Diff line change 1010 set_layer_name ,
1111)
1212from torch_tensorrt .dynamo .conversion .impl .cat import (
13- unify_trt_tensors as unify_trt_shape_tensors ,
13+ unify_and_concat_trt_tensors as unify_trt_shape_tensors ,
1414)
1515from torch_tensorrt .dynamo .conversion .impl .shape import (
1616 get_shape_with_dynamic_shape ,
You can’t perform that action at this time.
0 commit comments