Weight.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/requirements/abstract/SingleSolvingRequirement.h>
 
namespace tvm
{
 
namespace requirements
{
 
template<bool Lightweight = true>
class WeightBase : public abstract::SingleSolvingRequirement<double, Lightweight>
{
public:
  WeightBase() : abstract::SingleSolvingRequirement<double, Lightweight>(1.0, true) {}
 
  WeightBase(double alpha) : abstract::SingleSolvingRequirement<double, Lightweight>(alpha, false)
  {
    if(alpha < 0)
      throw std::runtime_error("weight must be non negative.");
  }
 
  TVM_DEFINE_LW_NON_LW_CONVERSION_OPERATORS(WeightBase, double, Lightweight)
};
 
using Weight = WeightBase<true>;
 
} // namespace requirements
 
} // namespace tvm