TVM  0.9.4
tvm::hint::internal::SubstitutionUnit Class Reference

#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
 

Detailed Description

A class to perform all the computations related to a group of dependent substitutions.

Constructor & Destructor Documentation

◆ SubstitutionUnit()

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.

Member Function Documentation

◆ additionalConstraints()

const std::vector<std::shared_ptr<constraint::BasicLinearConstraint> >& tvm::hint::internal::SubstitutionUnit::additionalConstraints ( ) const

Return the additional constraints to be added to the problem.

◆ additionalVariables()

const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::additionalVariables ( ) const

Return all the z variables (including the empty ones).

◆ otherVariables()

const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::otherVariables ( ) const

Return all the y variables.

◆ update()

void tvm::hint::internal::SubstitutionUnit::update ( )

Update the matrices and vectors used for the substitutions.

◆ variables()

const std::vector<VariablePtr>& tvm::hint::internal::SubstitutionUnit::variables ( ) const

Return the substituted variables

◆ variableSubstitutions()

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()


The documentation for this class was generated from the following file: