defaultLeastSquareSolver.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/solver/abstract/LeastSquareSolver.h>
 
namespace tvm::solver
{
class DefaultLSSolverFactory;
 
class TVM_DLLAPI DefaultLSSolverOptions
{
  TVM_ADD_NON_DEFAULT_OPTION(big_number, constant::big_number)
  TVM_ADD_NON_DEFAULT_OPTION(verbose, false)
 
public:
  using Factory = DefaultLSSolverFactory;
};
 
class TVM_DLLAPI DefaultLSSolverFactory : public abstract::LSSolverFactory
{
public:
  std::unique_ptr<abstract::LSSolverFactory> clone() const override;
 
  DefaultLSSolverFactory(const DefaultLSSolverOptions & options = {});
 
  std::unique_ptr<abstract::LeastSquareSolver> createSolver() const override;
 
private:
  DefaultLSSolverOptions options_;
};
} // namespace tvm::solver