Loss.h File Reference

#include <numeric>
#include "cker/Shape.h"
#include "cker/eigen/EigenSupport.h"
#include "cker/eigen/Utils.h"
#include "cker/eigen/xent_op.h"
#include "cker/operation/Helper/BCast.h"
#include "cker/train/Types.h"

Go to the source code of this file.

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

Functions
template<typename T >
T	nnfw::cker::train::square (T value)
template<typename T >
T	nnfw::cker::train::log_threshold ()
template<typename T >
void	nnfw::cker::train::MSE (const Shape &y_pred_shape, const T *y_pred_data, const Shape &y_true_shape, const T *y_true_data, const Shape &output_shape, T *output_data)
template<typename T >
void	nnfw::cker::train::MSEGrad (const Shape &y_pred_shape, const T *y_pred_data, const Shape &y_true_shape, const T *y_true_data, const Shape &grad_shape, T *grad_data, LossReductionType reduction_type)
template<typename T >
void	nnfw::cker::train::CategoricalCrossEntropy (const Shape &y_pred_shape, const T *y_pred_data, const Shape &y_true_shape, const T *y_true_data, const Shape &output_shape, T *output_data)
template<typename T >
void	nnfw::cker::train::CategoricalCrossEntropyGrad (const Shape &y_pred_shape, const T *y_pred_data, const Shape &y_true_shape, const T *y_true_data, const Shape &grad_shape, T *grad_data, LossReductionType reduction_type)
template<typename T >
void	nnfw::cker::train::CategoricalCrossEntropyWithLogits (const Shape &logits_shape, const T *logits_data, const Shape &y_true_shape, const T *y_true_data, const Shape &loss_out_shape, T *loss_out_data, const Shape &grad_shape, T *grad_data, LossReductionType reduction_type)