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 85e28ac commit fed0533Copy full SHA for fed0533
py/torch_tensorrt/dynamo/conversion/impl/cat.py
@@ -24,10 +24,10 @@ def cat(
24
trt_inputs = []
25
for i, each_input in enumerate(input):
26
if not isinstance(each_input, TRTTensor):
27
- each_input = get_trt_tensor(ctx, each_input, name + f"_tensor_{i}")
+ each_input = get_trt_tensor(ctx, each_input, f"{name}_tensor_{i}")
28
trt_inputs.append(each_input)
29
concat_layer = ctx.net.add_concatenation(trt_inputs)
30
dim = get_positive_dim(dim, len(input[0].shape))
31
concat_layer.axis = dim
32
- set_layer_name(concat_layer, target, name + "_gather", source_ir)
+ set_layer_name(concat_layer, target, f"{name}_gather", source_ir)
33
return concat_layer.get_output(0)
0 commit comments