|
TVM
0.9.4
|
#include <tvm/hint/internal/SubstitutionUnit.h>
Public Member Functions | |
| SubstitutionUnit (const std::vector< Substitution > &substitutionPool, const std::vector< std::vector< size_t >> &groups, const std::vector< size_t > order) | |
| void | update () |
| 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 |
A class to perform all the computations related to a group of dependent substitutions.
| tvm::hint::internal::SubstitutionUnit::SubstitutionUnit | ( | const std::vector< Substitution > & | substitutionPool, |
| const std::vector< std::vector< size_t >> & | groups, | ||
| const std::vector< size_t > | order | ||
| ) |
Build a SubstitutionUnit containing the substitutions from substitutionPool with indices in groups[i] for each i in order. Substitutions in a same groups[i] are merged as a single substitution.
| const std::vector<std::shared_ptr<constraint::BasicLinearConstraint> >& tvm::hint::internal::SubstitutionUnit::additionalConstraints | ( | ) | const |
Return the additional constraints to be added to the problem.
| const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::additionalVariables | ( | ) | const |
Return all the z variables (including the empty ones).
| const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::otherVariables | ( | ) | const |
Return all the y variables.
| void tvm::hint::internal::SubstitutionUnit::update | ( | ) |
Update the matrices and vectors used for the substitutions.
| const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::variables | ( | ) | const |
Return the substituted variables
| const std::vector<std::shared_ptr<function::BasicLinearFunction> >& tvm::hint::internal::SubstitutionUnit::variableSubstitutions | ( | ) | const |
Return the function giving the values of the substituted variables. The order of the vector corresponds to the order of the vector returned by variables()