Adam.h File Reference

#include "cker/eigen/training_ops.h"
#include "cker/eigen/EigenSupport.h"
#include <vector>

Go to the source code of this file.

Namespaces
namespace	nnfw
namespace	nnfw::cker
namespace	nnfw::cker::train

Functions
void	nnfw::cker::train::Adam (const Shape &trainable_shape, float *trainable_data, const Shape &grad_shape, const float *grad_data, const Shape &m_shape, float *m_data, const Shape &v_shape, float *v_data, float beta1_power, float beta2_power, float learning_rate, float beta1, float beta2, float epsilon, bool use_nesterov)