Go to the documentation of this file.00001
00002 #ifndef GRU_LAYER_H
00003 #define GRU_LAYER_H
00004
00005 #include "activations.h"
00006 #include "layer.h"
00007 #include "network.h"
00008
00009 layer make_gru_layer(int batch, int inputs, int outputs, int steps, int batch_normalize);
00010
00011 void forward_gru_layer(layer l, network_state state);
00012 void backward_gru_layer(layer l, network_state state);
00013 void update_gru_layer(layer l, int batch, float learning_rate, float momentum, float decay);
00014
00015 #ifdef GPU
00016 void forward_gru_layer_gpu(layer l, network_state state);
00017 void backward_gru_layer_gpu(layer l, network_state state);
00018 void update_gru_layer_gpu(layer l, int batch, float learning_rate, float momentum, float decay);
00019 void push_gru_layer(layer l);
00020 void pull_gru_layer(layer l);
00021 #endif
00022
00023 #endif
00024