Substitutions.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
#include <tvm/defs.h>
 
#include <tvm/graph/internal/DependencyGraph.h>
#include <tvm/hint/Substitution.h>
#include <tvm/hint/internal/SubstitutionUnit.h>
 
#include <vector>
 
namespace tvm
{
 
namespace hint
{
 
namespace internal
{
class TVM_DLLAPI Substitutions
{
public:
  void add(const Substitution & s);
 
  void remove(const Substitution & s);
 
  const std::vector<Substitution> & substitutions() const;
 
  bool uses(LinearConstraintPtr c) const;
 
  void finalize();
 
  void updateSubstitutions();
 
  void updateVariableValues() const;
 
  const std::vector<VariablePtr> & variables() const;
  const std::vector<std::shared_ptr<function::BasicLinearFunction>> & variableSubstitutions() const;
  const std::vector<VariablePtr> & additionalVariables() const;
  const std::vector<std::shared_ptr<constraint::BasicLinearConstraint>> & additionalConstraints() const;
  const std::vector<VariablePtr> & otherVariables() const;
  VariableVector substitute(const VariablePtr & x) const;
 
  Substitution const * getSubstitutionFor(const constraint::abstract::LinearConstraint & cstr);
 
private:
  std::vector<Substitution> substitutions_;
 
  tvm::graph::internal::DependencyGraph dependencies_;
 
  std::vector<SubstitutionUnit> units_;
 
  std::vector<VariablePtr> variables_;
 
  std::vector<std::shared_ptr<function::BasicLinearFunction>> varSubstitutions_;
 
  std::vector<VariablePtr> additionalVariables_;
 
  std::vector<VariablePtr> otherVariables_;
 
  std::vector<std::shared_ptr<constraint::BasicLinearConstraint>> additionalConstraints_;
 
  friend class SubstitutionTest;
};
 
} // namespace internal
 
} // namespace hint
 
} // namespace tvm