TVM  0.9.4
tvm::scheme::internal::WeightMultBase< DIAGONAL > Class Reference

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

Public Types

template<typename T >
using ReturnType = decltype(std::declval< Eigen::Ref< const Eigen::VectorXd > >().asDiagonal() *std::declval< T >())
 

Public Member Functions

 WeightMultBase (const Eigen::Ref< const Eigen::VectorXd > &d)
 
template<typename T >
ReturnType< T > applyWeightMult (const T &M)
 
decltype(double() *std::declval< Eigen::Ref< const Eigen::VectorXd >>()) applyWeightMult (const double &d)
 

Detailed Description

Specialization for DIAGONAL

Member Typedef Documentation

◆ ReturnType

template<typename T >
using tvm::scheme::internal::WeightMultBase< DIAGONAL >::ReturnType = decltype(std::declval<Eigen::Ref<const Eigen::VectorXd> >().asDiagonal() * std::declval<T>())

Constructor & Destructor Documentation

◆ WeightMultBase()

tvm::scheme::internal::WeightMultBase< DIAGONAL >::WeightMultBase ( const Eigen::Ref< const Eigen::VectorXd > &  d)
inline

Member Function Documentation

◆ applyWeightMult() [1/2]

decltype(double() * std::declval<Eigen::Ref<const Eigen::VectorXd>>()) tvm::scheme::internal::WeightMultBase< DIAGONAL >::applyWeightMult ( const double &  d)
inline

Diagonal * constant vector case

◆ applyWeightMult() [2/2]

template<typename T >
ReturnType<T> tvm::scheme::internal::WeightMultBase< DIAGONAL >::applyWeightMult ( const T &  M)
inline

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