crnn_layer.h 649 B

123456789101112131415161718192021222324
  1. #ifndef CRNN_LAYER_H
  2. #define CRNN_LAYER_H
  3. #include "activations.h"
  4. #include "layer.h"
  5. #include "network.h"
  6. 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);
  7. void forward_crnn_layer(layer l, network net);
  8. void backward_crnn_layer(layer l, network net);
  9. void update_crnn_layer(layer l, update_args a);
  10. #ifdef GPU
  11. void forward_crnn_layer_gpu(layer l, network net);
  12. void backward_crnn_layer_gpu(layer l, network net);
  13. void update_crnn_layer_gpu(layer l, update_args a);
  14. void push_crnn_layer(layer l);
  15. void pull_crnn_layer(layer l);
  16. #endif
  17. #endif