magmadnn::optimizer::Optimizer< T > Class Template Reference

Inheritance diagram for magmadnn::optimizer::Optimizer< T >:

Collaboration diagram for magmadnn::optimizer::Optimizer< T >:

Public Member Functions
virtual void	minimize (op::Operation< T > *obj_func, const std::vector< op::Operation< T > *> &wrt)=0
virtual std::string	get_name ()

Protected Member Functions
virtual void	update (op::Operation< T > *var, Tensor< T > *grad)=0

Protected Attributes
op::Operation< T > *	_obj_func
std::string	_name = "Generic Optimizer"

---

The documentation for this class was generated from the following file:

• include/optimizer/optimizer.h