SGD.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::GradientDescent (const Shape &output_shape, float *output_data, const Shape &grad_shape, const float *grad_data, float learning_rate)