lstm_layer.c File Reference

#include "lstm_layer.h"
#include "connected_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

Include dependency graph for lstm_layer.c:

Go to the source code of this file.

Functions
layer	make_lstm_layer (int batch, int inputs, int outputs, int steps, int batch_normalize, int adam)
void	update_lstm_layer (layer l, update_args a)
void	forward_lstm_layer (layer l, network state)
void	backward_lstm_layer (layer l, network state)

Function Documentation

backward_lstm_layer()

void backward_lstm_layer	(	layer	l,
		network	state
	)

Definition at line 242 of file lstm_layer.c.

forward_lstm_layer()

void forward_lstm_layer	(	layer	l,
		network	state
	)

Definition at line 156 of file lstm_layer.c.

make_lstm_layer()

layer make_lstm_layer	(	int	batch,
		int	inputs,
		int	outputs,
		int	steps,
		int	batch_normalize,
		int	adam
	)

Definition at line 29 of file lstm_layer.c.

update_lstm_layer()

void update_lstm_layer	(	layer	l,
		update_args	a
	)

Definition at line 144 of file lstm_layer.c.