|
TVM
0.9.4
|
#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) |
Specialization for DIAGONAL
| using tvm::scheme::internal::WeightMultBase< DIAGONAL >::ReturnType = decltype(std::declval<Eigen::Ref<const Eigen::VectorXd> >().asDiagonal() * std::declval<T>()) |
|
inline |
|
inline |
Diagonal * constant vector case
|
inline |