batchnorm_layer.c File Reference

#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "blas.h"
#include <stdio.h>

Include dependency graph for batchnorm_layer.c:

Go to the source code of this file.

Functions
layer	make_batchnorm_layer (int batch, int w, int h, int c)
void	backward_scale_cpu (float *x_norm, float *delta, int batch, int n, int size, float *scale_updates)
void	mean_delta_cpu (float *delta, float *variance, int batch, int filters, int spatial, float *mean_delta)
void	variance_delta_cpu (float *x, float *delta, float *mean, float *variance, int batch, int filters, int spatial, float *variance_delta)
void	normalize_delta_cpu (float *x, float *mean, float *variance, float *mean_delta, float *variance_delta, int batch, int filters, int spatial, float *delta)
void	resize_batchnorm_layer (layer *layer, int w, int h)
void	forward_batchnorm_layer (layer l, network net)
void	backward_batchnorm_layer (layer l, network net)

Function Documentation

backward_batchnorm_layer()

void backward_batchnorm_layer	(	layer	l,
		network	net
	)

Definition at line 157 of file batchnorm_layer.c.

backward_scale_cpu()

void backward_scale_cpu	(	float *	x_norm,
		float *	delta,
		int	batch,
		int	n,
		int	size,
		float *	scale_updates
	)

Definition at line 72 of file batchnorm_layer.c.

forward_batchnorm_layer()

void forward_batchnorm_layer	(	layer	l,
		network	net
	)

Definition at line 135 of file batchnorm_layer.c.

make_batchnorm_layer()

layer make_batchnorm_layer	(	int	batch,
		int	w,
		int	h,
		int	c
	)

Definition at line 6 of file batchnorm_layer.c.

mean_delta_cpu()

void mean_delta_cpu	(	float *	delta,
		float *	variance,
		int	batch,
		int	filters,
		int	spatial,
		float *	mean_delta
	)

Definition at line 87 of file batchnorm_layer.c.

normalize_delta_cpu()

void normalize_delta_cpu	(	float *	x,
		float *	mean,
		float *	variance,
		float *	mean_delta,
		float *	variance_delta,
		int	batch,
		int	filters,
		int	spatial,
		float *	delta
	)

Definition at line 117 of file batchnorm_layer.c.

resize_batchnorm_layer()

void resize_batchnorm_layer	(	layer *	layer,
		int	w,
		int	h
	)

Definition at line 130 of file batchnorm_layer.c.

variance_delta_cpu()

void variance_delta_cpu	(	float *	x,
		float *	delta,
		float *	mean,
		float *	variance,
		int	batch,
		int	filters,
		int	spatial,
		float *	variance_delta
	)

Definition at line 102 of file batchnorm_layer.c.