Constraint.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/constraint/enums.h>
#include <tvm/constraint/internal/RHSVectors.h>
#include <tvm/graph/abstract/OutputSelector.h>
#include <tvm/internal/FirstOrderProvider.h>
 
#include <Eigen/Core>
 
#include <memory>
 
namespace tvm
{
 
namespace constraint
{
 
namespace abstract
{
 
class TVM_DLLAPI Constraint : public tvm::internal::ObjWithId,
                              public graph::abstract::OutputSelector<Constraint, tvm::internal::FirstOrderProvider>
{
public:
  SET_OUTPUTS(Constraint, L, U, E)
 
  
  virtual const Eigen::VectorXd & l() const;
  virtual const Eigen::VectorXd & u() const;
  virtual const Eigen::VectorXd & e() const;
 
  Type type() const;
 
  bool isEquality() const;
 
  RHS rhs() const;
 
protected:
  Constraint(Type ct, RHS cr, int m = 0);
 
  void resizeCache() override;
 
  Eigen::VectorXd & lRef();
  Eigen::VectorXd & uRef();
  Eigen::VectorXd & eRef();
 
  internal::RHSVectors vectors_;
 
private:
  Type cstrType_;     // The constraint type
  RHS constraintRhs_; // The rhs convention
};
 
inline const Eigen::VectorXd & Constraint::l() const { return vectors_.l(); }
 
inline const Eigen::VectorXd & Constraint::u() const { return vectors_.u(); }
 
inline const Eigen::VectorXd & Constraint::e() const { return vectors_.e(); }
 
inline Eigen::VectorXd & Constraint::lRef() { return vectors_.l(); }
 
inline Eigen::VectorXd & Constraint::uRef() { return vectors_.u(); }
 
inline Eigen::VectorXd & Constraint::eRef() { return vectors_.e(); }
 
} // namespace abstract
 
} // namespace constraint
 
} // namespace tvm