TVM  0.9.4
tvm::hint::internal::DiagonalCalculator::Impl Class Reference

#include <tvm/hint/internal/DiagonalCalculator.h>

Inheritance diagram for tvm::hint::internal::DiagonalCalculator::Impl:
Collaboration diagram for tvm::hint::internal::DiagonalCalculator::Impl:

Public Member Functions

 Impl (const std::vector< LinearConstraintPtr > &cstr, const std::vector< VariablePtr > &x, int rank, Eigen::DenseIndex first, Eigen::DenseIndex size)
 
 Impl (const std::vector< LinearConstraintPtr > &cstr, const std::vector< VariablePtr > &x, int rank, const std::vector< Eigen::DenseIndex > &nnzRows, const std::vector< Eigen::DenseIndex > &zeroRows)
 
virtual void update_ () override
 
virtual void premultiplyByASharpAndSTranspose_ (MatrixRef outA, MatrixRef outS, const MatrixConstRef &in, bool minus) const override
 
virtual void postMultiplyByN_ (MatrixRef out, const MatrixConstRef &in, bool add) const override
 
- Public Member Functions inherited from tvm::hint::abstract::SubstitutionCalculatorImpl
virtual ~SubstitutionCalculatorImpl ()=default
 
void update ()
 
void premultiplyByASharpAndSTranspose (MatrixRef outA, MatrixRef outS, const MatrixConstRef &in, bool minus) const
 
void postMultiplyByN (MatrixRef out, const MatrixConstRef &in, bool add=false) const
 
void postMultiplyByN (MatrixRef out, const MatrixConstRef &in, Range r, bool add=false) const
 
const Eigen::MatrixXd & N () const
 
Eigen::DenseIndex m () const
 
Eigen::DenseIndex n () const
 
Eigen::DenseIndex r () const
 

Additional Inherited Members

- Protected Member Functions inherited from tvm::hint::abstract::SubstitutionCalculatorImpl
 SubstitutionCalculatorImpl (const std::vector< LinearConstraintPtr > &cstr, const std::vector< VariablePtr > &x, int rank)
 
void constant (bool c)
 
bool constant () const
 
MatrixConstRef A () const
 
bool isSimple () const
 
virtual void postMultiplyByN_ (MatrixRef out, const MatrixConstRef &in, Range r, bool add) const
 
void fillA ()
 
- Protected Attributes inherited from tvm::hint::abstract::SubstitutionCalculatorImpl
Eigen::MatrixXd N_
 
std::vector< LinearConstraintPtrconstraints_
 
VariableVector variables_
 

Constructor & Destructor Documentation

◆ Impl() [1/2]

tvm::hint::internal::DiagonalCalculator::Impl::Impl ( const std::vector< LinearConstraintPtr > &  cstr,
const std::vector< VariablePtr > &  x,
int  rank,
Eigen::DenseIndex  first,
Eigen::DenseIndex  size 
)

◆ Impl() [2/2]

tvm::hint::internal::DiagonalCalculator::Impl::Impl ( const std::vector< LinearConstraintPtr > &  cstr,
const std::vector< VariablePtr > &  x,
int  rank,
const std::vector< Eigen::DenseIndex > &  nnzRows,
const std::vector< Eigen::DenseIndex > &  zeroRows 
)

Member Function Documentation

◆ postMultiplyByN_()

virtual void tvm::hint::internal::DiagonalCalculator::Impl::postMultiplyByN_ ( MatrixRef  out,
const MatrixConstRef in,
bool  add 
) const
overridevirtual

◆ premultiplyByASharpAndSTranspose_()

virtual void tvm::hint::internal::DiagonalCalculator::Impl::premultiplyByASharpAndSTranspose_ ( MatrixRef  outA,
MatrixRef  outS,
const MatrixConstRef in,
bool  minus 
) const
overridevirtual

◆ update_()

virtual void tvm::hint::internal::DiagonalCalculator::Impl::update_ ( )
overridevirtual

Handle for the derived class to perform the computations in update()

Implements tvm::hint::abstract::SubstitutionCalculatorImpl.


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