Skip to content

Commit 9b041ab

Browse files
hossein1387facebook-github-bot
authored andcommitted
aten.leakyrelu.default Op registery (pytorch#8227)
Summary: leaky_relu op registery was missing from D68688186. Differential Revision: D69199644
1 parent b66c319 commit 9b041ab

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/vulkan/op_registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,7 @@ def register_binary_op(features: OpFeatures):
276276
exir_ops.edge.aten.sqrt.default,
277277
exir_ops.edge.aten.rsqrt.default,
278278
exir_ops.edge.aten.tanh.default,
279+
exir_ops.edge.aten.leaky_relu.default,
279280
]
280281
)
281282
def register_unary_op(features: OpFeatures):

0 commit comments

Comments
 (0)