TVM  0.9.4
tvm::scheme::internal::MatrixAssignment Class Reference

#include <tvm/scheme/internal/MatrixAssignment.h>

Collaboration diagram for tvm::scheme::internal::MatrixAssignment:

Public Types

using MatrixFunction = MatrixRef(AssignmentTarget::*)(int, int) const
 

Public Member Functions

void updateTarget (const AssignmentTarget &target)
 
void updateMapping (const VariableVector &newVar, const AssignmentTarget &target, bool updateMatrixTarget)
 

Public Attributes

CompiledAssignmentWrapper< Eigen::MatrixXd > assignment
 
Variablex
 
Range colRange
 
MatrixFunction getTargetMatrix
 

Detailed Description

A structure grouping a matrix assignment and some of the elements that defined it.

Member Typedef Documentation

◆ MatrixFunction

using tvm::scheme::internal::MatrixAssignment::MatrixFunction = MatrixRef (AssignmentTarget::*)(int, int) const

Pointer type to a method of AssignmentTarget returning a matrix block. It is used to make a selection between A(), AFirstHalf() and ASecondHalf().

Member Function Documentation

◆ updateMapping()

void tvm::scheme::internal::MatrixAssignment::updateMapping ( const VariableVector newVar,
const AssignmentTarget target,
bool  updateMatrixTarget 
)

Update the column mapping of this assignment, based on the new variable layout specified by newVar. If updateMatrixTarget is true, the output matrix is effectively updated (if not, only colRange is changed, but the change is not reflected in the actual compiled assignment).

◆ updateTarget()

void tvm::scheme::internal::MatrixAssignment::updateTarget ( const AssignmentTarget target)

Effectively update the output matrix of the underlying compiled assignment.

Member Data Documentation

◆ assignment

CompiledAssignmentWrapper<Eigen::MatrixXd> tvm::scheme::internal::MatrixAssignment::assignment

◆ colRange

Range tvm::scheme::internal::MatrixAssignment::colRange

◆ getTargetMatrix

MatrixFunction tvm::scheme::internal::MatrixAssignment::getTargetMatrix

◆ x

Variable* tvm::scheme::internal::MatrixAssignment::x

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