nnfw::cker::training_ops::functor::ApplyAdam< Device, T > Struct Template Reference

#include <training_ops.h>

Public Member Functions
void	operator() (const Device &d, typename TTypes< T >::Flat var, typename TTypes< T >::Flat m, typename TTypes< T >::Flat v, typename TTypes< T >::ConstScalar beta1_power, typename TTypes< T >::ConstScalar beta2_power, typename TTypes< T >::ConstScalar lr, typename TTypes< T >::ConstScalar beta1, typename TTypes< T >::ConstScalar beta2, typename TTypes< T >::ConstScalar epsilon, typename TTypes< T >::ConstFlat grad, bool use_nesterov)

Detailed Description

template<typename Device, typename T>
struct nnfw::cker::training_ops::functor::ApplyAdam< Device, T >

Definition at line 37 of file training_ops.h.

Member Function Documentation

operator()()

template<typename Device , typename T >

void nnfw::cker::training_ops::functor::ApplyAdam< Device, T >::operator()	(	const Device &	d,
		typename TTypes< T >::Flat	var,
		typename TTypes< T >::Flat	m,
		typename TTypes< T >::Flat	v,
		typename TTypes< T >::ConstScalar	beta1_power,
		typename TTypes< T >::ConstScalar	beta2_power,
		typename TTypes< T >::ConstScalar	lr,
		typename TTypes< T >::ConstScalar	beta1,
		typename TTypes< T >::ConstScalar	beta2,
		typename TTypes< T >::ConstScalar	epsilon,
		typename TTypes< T >::ConstFlat	grad,
		bool	use_nesterov
	)

---

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

• compute/cker/include/cker/eigen/training_ops.h