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 049b810 commit 015bf19Copy full SHA for 015bf19
py/torch_tensorrt/dynamo/conversion/impl/select.py
@@ -258,16 +258,15 @@ def index(
258
else:
259
dim_tensor_shape_mult_d1 = transpose_tensor_shape[i]
260
261
- if isinstance(dim_tensor_shape_mult_d1, TRTTensor):
262
- mult_d1 = convert_binary_elementwise(
263
- ctx,
264
- target,
265
- source_ir,
266
- name + f"_shape_{i}",
267
- trt.ElementWiseOperation.PROD,
268
- mult_d1,
269
- dim_tensor_shape_mult_d1,
270
- )
+ mult_d1 = convert_binary_elementwise(
+ ctx,
+ target,
+ source_ir,
+ name + f"_shape_{i}",
+ trt.ElementWiseOperation.PROD,
+ mult_d1,
+ dim_tensor_shape_mult_d1,
+ )
271
272
concat_tensor_layer = ctx.net.add_concatenation(
273
[
0 commit comments