Skip to content

Commit 3bfa4e0

Browse files
committed
changing function name
1 parent 3fcf398 commit 3bfa4e0

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
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,

py/torch_tensorrt/dynamo/conversion/impl/upsample.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
set_layer_name,
1111
)
1212
from 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
)
1515
from torch_tensorrt.dynamo.conversion.impl.shape import (
1616
get_shape_with_dynamic_shape,

0 commit comments

Comments
 (0)