|
TVM
0.9.4
|
#include <tvm/hint/internal/DiagonalCalculator.h>


Classes | |
| class | Impl |
Public Member Functions | |
| DiagonalCalculator (Eigen::DenseIndex first=0, Eigen::DenseIndex size=-1) | |
| DiagonalCalculator (const std::vector< Eigen::DenseIndex > &nnzRows, const std::vector< Eigen::DenseIndex > &zeroRows={}) | |
Public Member Functions inherited from tvm::hint::abstract::SubstitutionCalculator | |
| virtual | ~SubstitutionCalculator ()=default |
| std::unique_ptr< SubstitutionCalculatorImpl > | impl (const std::vector< LinearConstraintPtr > &cstr, const std::vector< VariablePtr > &x, int rank) const |
Protected Member Functions | |
| std::unique_ptr< abstract::SubstitutionCalculatorImpl > | impl_ (const std::vector< LinearConstraintPtr > &cstr, const std::vector< VariablePtr > &x, int rank) const |
A calculator for all matrices that are the row selection of a diagonal matrix. The structure of the matrix (position of non-zero elements) is assumed to be constant.
| tvm::hint::internal::DiagonalCalculator::DiagonalCalculator | ( | Eigen::DenseIndex | first = 0, |
| Eigen::DenseIndex | size = -1 |
||
| ) |
Continuous rows of a diagonal matrix starting at row first of this matrix and finishing at row first + size (not included). If size = -1, finishes at the last row.
| tvm::hint::internal::DiagonalCalculator::DiagonalCalculator | ( | const std::vector< Eigen::DenseIndex > & | nnzRows, |
| const std::vector< Eigen::DenseIndex > & | zeroRows = {} |
||
| ) |
Non-continuous rows of a diagonal matrix
| nnzRows | indices of the rows with a non-zero element. |
| zeroRows | indices of the rows (if any) with only zero |
Examples: Knowing that it describes a matrix with 6 columns, nnzRows = {0,2,3}, zeroRows = {} corresponds to the shape x 0 0 0 0 0 0 0 x 0 0 0 0 0 0 x 0 0
while nnzRows = {0,2,3}, zeroRows = {1,4} corresponds to x 0 0 0 0 0 0 0 0 0 0 0 0 0 x 0 0 0 0 0 0 x 0 0 0 0 0 0 0 0
|
protectedvirtual |
Implements tvm::hint::abstract::SubstitutionCalculator.