LinearFunction.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/function/abstract/Function.h>
 
#include <Eigen/Core>
 
#include <initializer_list>
#include <memory>
 
namespace tvm
{
 
namespace function
{
 
namespace abstract
{
 
class TVM_DLLAPI LinearFunction : public Function
{
public:
  SET_OUTPUTS(LinearFunction, B)
  SET_UPDATES(LinearFunction, Value, Velocity)
 
  void updateValue();
  void updateVelocity();
  void resizeCache() override;
  const tvm::internal::VectorWithProperties & b() const { return b_; }
 
protected:
  LinearFunction(int m);
  virtual void updateValue_();
  virtual void updateVelocity_();
  void setDerivativesToZero();
 
  tvm::internal::VectorWithProperties b_;
};
 
} // namespace abstract
 
} // namespace function
 
} // namespace tvm