Go to the documentation of this file.00001
00002 #ifndef CRNN_LAYER_H
00003 #define CRNN_LAYER_H
00004
00005 #include "activations.h"
00006 #include "layer.h"
00007 #include "network.h"
00008
00009 layer make_crnn_layer(int batch, int h, int w, int c, int hidden_filters, int output_filters, int steps, ACTIVATION activation, int batch_normalize);
00010
00011 void forward_crnn_layer(layer l, network_state state);
00012 void backward_crnn_layer(layer l, network_state state);
00013 void update_crnn_layer(layer l, int batch, float learning_rate, float momentum, float decay);
00014
00015 #ifdef GPU
00016 void forward_crnn_layer_gpu(layer l, network_state state);
00017 void backward_crnn_layer_gpu(layer l, network_state state);
00018 void update_crnn_layer_gpu(layer l, int batch, float learning_rate, float momentum, float decay);
00019 void push_crnn_layer(layer l);
00020 void pull_crnn_layer(layer l);
00021 #endif
00022
00023 #endif
00024