implementing addReluConstraint()

-implemented method that adds the Relu constraints given two variables -avoids exposing PiecewiseLinearConstraint to the shared library
NeuralNetworkVerification · Mar 10, 2018 · 75eb4f3 · 75eb4f3
1 parent 87d641a
commit 75eb4f3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/engine/InputQuery.cpp b/src/engine/InputQuery.cpp
@@ -14,6 +14,7 @@
 #include "MStringf.h"
 #include "ReluplexError.h"
 #include "FloatUtils.h"
+#include "ReluConstraint.h"
 
 InputQuery::InputQuery()
 {
@@ -117,6 +118,12 @@ double InputQuery::getSolutionValue( unsigned variable ) const
     return _solution.get( variable );
 }
 
+void InputQuery::addReluConstraint(unsigned var1, unsigned var2)
+{
+    PiecewiseLinearConstraint* r = new ReluConstraint(var1, var2);
+    addPiecewiseLinearConstraint(r);
+}
+
 void InputQuery::addPiecewiseLinearConstraint( PiecewiseLinearConstraint *constraint )
 {
     _plConstraints.append( constraint );

diff --git a/src/engine/InputQuery.h b/src/engine/InputQuery.h
@@ -42,6 +42,7 @@ class InputQuery
     const List<Equation> &getEquations() const;
 	List<Equation> &getEquations();
 
+    void addReluConstraint(unsigned var1, unsigned var2);
     void addPiecewiseLinearConstraint( PiecewiseLinearConstraint *constraint );
     const List<PiecewiseLinearConstraint *> &getPiecewiseLinearConstraints() const;
 	List<PiecewiseLinearConstraint *> &getPiecewiseLinearConstraints();