softmax_layer.c File Reference

#include "softmax_layer.h"
#include "blas.h"
#include "cuda.h"
#include <float.h>
#include <math.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>

Include dependency graph for softmax_layer.c:

Go to the source code of this file.

Functions
softmax_layer	make_softmax_layer (int batch, int inputs, int groups)
void	forward_softmax_layer (const softmax_layer l, network net)
void	backward_softmax_layer (const softmax_layer l, network net)

Function Documentation

backward_softmax_layer()

void backward_softmax_layer	(	const softmax_layer	l,
		network	net
	)

Definition at line 59 of file softmax_layer.c.

forward_softmax_layer()

void forward_softmax_layer	(	const softmax_layer	l,
		network	net
	)

Definition at line 39 of file softmax_layer.c.

make_softmax_layer()

softmax_layer make_softmax_layer	(	int	batch,
		int	inputs,
		int	groups
	)

Definition at line 11 of file softmax_layer.c.