SoftMax.h File Reference

#include "cker/Shape.h"
#include "cker/eigen/Utils.h"

Go to the source code of this file.

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

Functions
void	nnfw::cker::train::SoftMaxGrad (const Shape &output_shape, const float *output_data, const Shape &incoming_shape, const float *incoming_data, const Shape &grad_shape, float *grad_data)