Substitution.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
#include <tvm/defs.h>
 
#include <tvm/hint/abstract/SubstitutionCalculator.h>
#include <tvm/hint/internal/AutoCalculator.h>
 
#include <vector>
 
namespace tvm
{
 
namespace hint
{
 
namespace abstract
{
class SubstitutionCalculatorImpl;
}
 
class TVM_DLLAPI Substitution
{
public:
  Substitution(LinearConstraintPtr cstr,
               VariablePtr x,
               int rank = constant::fullRank,
               const abstract::SubstitutionCalculator & calc = internal::AutoCalculator());
  Substitution(const std::vector<LinearConstraintPtr> & cstr,
               VariablePtr x,
               int rank = constant::fullRank,
               const abstract::SubstitutionCalculator & calc = internal::AutoCalculator());
  Substitution(LinearConstraintPtr cstr,
               std::vector<VariablePtr> & x,
               int rank = constant::fullRank,
               const abstract::SubstitutionCalculator & calc = internal::AutoCalculator());
  Substitution(const std::vector<LinearConstraintPtr> & cstr,
               const std::vector<VariablePtr> & x,
               int rank = constant::fullRank,
               const abstract::SubstitutionCalculator & calc = internal::AutoCalculator());
 
  int rank() const;
  int m() const;
  const std::vector<LinearConstraintPtr> & constraints() const;
  const std::vector<VariablePtr> & variables() const;
  bool isSimple() const;
 
  std::shared_ptr<abstract::SubstitutionCalculatorImpl> calculator() const;
 
private:
  void check() const;
 
  int rank_;
  int m_;
  std::vector<LinearConstraintPtr> constraints_;
  std::vector<VariablePtr> x_;
 
  std::shared_ptr<abstract::SubstitutionCalculatorImpl> calculator_;
};
} // namespace hint
 
} // namespace tvm