Skip to content

Commit 68c3184

Browse files
committed
resolve conflicts
1 parent 9a1c426 commit 68c3184

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -422,6 +422,24 @@ def aten_ops_clone(
422422
)
423423

424424

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+
425443
@dynamo_tensorrt_converter(torch.ops.aten.amax.default)
426444
def aten_ops_amax(
427445
network: TRTNetwork,

0 commit comments

Comments
 (0)