added class frameworks

MekAkUActOR · Nov 20, 2022 · bf38cd0 · bf38cd0
1 parent 97e4813
commit bf38cd0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/code/verifier.py b/code/verifier.py
@@ -106,6 +106,22 @@ def _get_ub(self, expr_w, expr_b):
             )
         else:
             return self.prev_layer._get_ub(res_w, res_b)
+
+class DPReLU(torch.nn.Module, DPBackSubstitution):
+    """ DeepPoly transformer for ReLU layer """
+    def __init__(self) -> None:
+        super(DPReLU, self).__init__()
+
+    def forward(self):
+        pass
+
+class DPLinear(torch.nn.Module, DPBackSubstitution):
+    """ DeepPoly transformer for affine layer """
+    def __init__(self) -> None:
+        super().__init__()
+
+    def forward(self):
+        pass
 
 
 def negative_only(w):