Add leaky relu metatype (#1005)

openvinotoolkit · Nov 8, 2021 · 53e8a64 · 53e8a64
1 parent 6f91efa
commit 53e8a64
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nncf/torch/pruning/operations.py b/nncf/torch/pruning/operations.py
@@ -43,6 +43,7 @@
     MulMetatype,
     PTOutputNoopMetatype,
     PRELUMetatype,
+    LeakyRELUMetatype,
     RELUMetatype,
     SigmoidMetatype,
     SILUMetatype,
@@ -105,7 +106,7 @@ class PTOutputPruningOp(OutputPruningOp, PTPruner):
 @PT_PRUNING_OPERATOR_METATYPES.register('identity_mask_propagation')
 class PTIdentityMaskForwardPruningOp(IdentityMaskForwardPruningOp, PTPruner):
     subtypes = [HardTanhMetatype, TanhMetatype, RELUMetatype, PRELUMetatype, ELUMetatype, GELUMetatype, SigmoidMetatype,
-                SoftmaxMetatype, AvgPool2dMetatype, MaxPool2dMetatype, DropoutMetatype, SILUMetatype]
+                SoftmaxMetatype, AvgPool2dMetatype, MaxPool2dMetatype, DropoutMetatype, SILUMetatype, LeakyRELUMetatype]
     additional_types = ['h_sigmoid', 'h_swish', 'RELU']