My Project
cuda_mlp
CudaGD
cuda_mlp::CudaGD Member List
This is the complete list of members for
cuda_mlp::CudaGD
, including all inherited members.
c1_
cuda_mlp::CudaMinimizerBase
protected
CudaGD
(CublasHandle &handle)
cuda_mlp::CudaGD
inline
explicit
CudaMinimizerBase
(CublasHandle &handle)
cuda_mlp::CudaMinimizerBase
inline
explicit
handle_
cuda_mlp::CudaMinimizerBase
protected
iterations
() const noexcept
cuda_mlp::CudaMinimizerBase
inline
IterHook
typedef
cuda_mlp::CudaMinimizerBase
last_iterations_
cuda_mlp::CudaMinimizerBase
protected
LossGradFun
typedef
cuda_mlp::CudaMinimizerBase
max_iters_
cuda_mlp::CudaMinimizerBase
protected
max_line_iters_
cuda_mlp::CudaMinimizerBase
protected
recorder_
cuda_mlp::CudaMinimizerBase
protected
rho_
cuda_mlp::CudaMinimizerBase
protected
setLearningRate
(CudaScalar lr)
cuda_mlp::CudaGD
inline
setLineSearchParams
(int max_iters, CudaScalar c1, CudaScalar rho)
cuda_mlp::CudaMinimizerBase
inline
setMaxIterations
(int iters)
cuda_mlp::CudaMinimizerBase
inline
setMomentum
(CudaScalar momentum)
cuda_mlp::CudaGD
inline
setRecorder
(::IterationRecorder< CudaBackend > *recorder)
cuda_mlp::CudaMinimizerBase
inline
setTolerance
(CudaScalar tol)
cuda_mlp::CudaMinimizerBase
inline
solve
(int n, CudaScalar *params, const CudaScalar *input, const CudaScalar *target, int total_samples, const LossGradFun &loss_grad) override
cuda_mlp::CudaGD
inline
virtual
tol_
cuda_mlp::CudaMinimizerBase
protected
~CudaMinimizerBase
()=default
cuda_mlp::CudaMinimizerBase
virtual
Generated by
1.9.1