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 9a1c426 commit 68c3184Copy full SHA for 68c3184
py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -422,6 +422,24 @@ def aten_ops_clone(
422
)
423
424
425
+@dynamo_tensorrt_converter(torch.ops.aten.expand.default)
426
+def aten_ops_expand(
427
+ network: TRTNetwork,
428
+ target: Target,
429
+ args: Tuple[Argument, ...],
430
+ kwargs: Dict[str, Argument],
431
+ name: str,
432
+) -> Union[TRTTensor, Sequence[TRTTensor]]:
433
+ return impl.slice.expand(
434
+ network,
435
+ target,
436
+ SourceIR.ATEN,
437
+ name,
438
+ args[0],
439
+ args[1],
440
+ )
441
+
442
443
@dynamo_tensorrt_converter(torch.ops.aten.amax.default)
444
def aten_ops_amax(
445
network: TRTNetwork,
0 commit comments