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 3d2ada6 commit 4ad2791Copy full SHA for 4ad2791
py/torch_tensorrt/dynamo/conversion/impl/select.py
@@ -87,9 +87,9 @@ def gather(
87
indices_tensor = []
88
89
for i, ind in enumerate(index):
90
- indices_tensor.append(get_trt_tensor(
91
- ctx, ind, name + f"_parameter_to_fp32_tensor_{i}"
92
- ))
+ indices_tensor.append(
+ get_trt_tensor(ctx, ind, name + f"_parameter_to_fp32_tensor_{i}")
+ )
93
gather_layer = ctx.net.add_gather(input, indices_tensor, dim)
94
set_layer_name(gather_layer, target, name + "_gather", source_ir)
95
return gather_layer.get_output(0)
0 commit comments