IdentityFunction.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/function/BasicLinearFunction.h>
 
namespace tvm
{
 
namespace function
{
 
class TVM_DLLAPI IdentityFunction : public BasicLinearFunction
{
public:
  IdentityFunction(VariablePtr x);
 
protected:
  void updateValue_() override;
  void updateVelocity_() override;
 
private:
  void A(const MatrixConstRef & A,
         const Variable & x,
         const tvm::internal::MatrixProperties & p = tvm::internal::MatrixProperties()) override;
  void A(const MatrixConstRef & A,
         const tvm::internal::MatrixProperties & p = tvm::internal::MatrixProperties()) override;
  void b(const VectorConstRef & b, const tvm::internal::MatrixProperties &) override;
};
 
} // namespace function
 
} // namespace tvm