Go to the documentation of this file.00001
00002 #ifndef RNN_LAYER_H
00003 #define RNN_LAYER_H
00004
00005 #include "activations.h"
00006 #include "layer.h"
00007 #include "network.h"
00008 #define USET
00009
00010 layer make_rnn_layer(int batch, int inputs, int hidden, int outputs, int steps, ACTIVATION activation, int batch_normalize, int log);
00011
00012 void forward_rnn_layer(layer l, network_state state);
00013 void backward_rnn_layer(layer l, network_state state);
00014 void update_rnn_layer(layer l, int batch, float learning_rate, float momentum, float decay);
00015
00016 #ifdef GPU
00017 void forward_rnn_layer_gpu(layer l, network_state state);
00018 void backward_rnn_layer_gpu(layer l, network_state state);
00019 void update_rnn_layer_gpu(layer l, int batch, float learning_rate, float momentum, float decay);
00020 void push_rnn_layer(layer l);
00021 void pull_rnn_layer(layer l);
00022 #endif
00023
00024 #endif
00025