softmax.h

Go to the documentation of this file.

#ifndef __NN_SOFTMAX_H
#define __NN_SOFTMAX_H
 
#include <stddef.h>
#include <math.h>
 
#include "float16.h"
 
 
void NN_softmax_f16(size_t n,
    float16_t *y, size_t incy,
    const float16_t *x, size_t incx
    );
 
void NN_softmax_f32(size_t n,
    float *y, size_t incy,
    const float *x, size_t incx
    );
 
 
#endif // __NN_SOFTMAX_H