Skip to content

Commit 95c2633

Browse files
committed
decompose prelu
1 parent d8461e6 commit 95c2633

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@
105105
aten.norm,
106106
aten.ones,
107107
aten.ones_like,
108+
aten._prelu_kernel,
108109
aten._prelu_kernel_backward,
109110
aten._reshape_alias,
110111
aten.rad2deg,

0 commit comments

Comments
 (0)