#include "rnn_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>
Go to the source code of this file.
Functions | |
| void | backward_rnn_layer (layer l, network_state state) |
| void | forward_rnn_layer (layer l, network_state state) |
| static void | increment_layer (layer *l, int steps) |
| layer | make_rnn_layer (int batch, int inputs, int hidden, int outputs, int steps, ACTIVATION activation, int batch_normalize, int log) |
| void | update_rnn_layer (layer l, int batch, float learning_rate, float momentum, float decay) |
| void backward_rnn_layer | ( | layer | l, |
| network_state | state | ||
| ) |
Definition at line 124 of file rnn_layer.c.
| void forward_rnn_layer | ( | layer | l, |
| network_state | state | ||
| ) |
Definition at line 83 of file rnn_layer.c.
| static void increment_layer | ( | layer * | l, |
| int | steps | ||
| ) | [static] |
Definition at line 13 of file rnn_layer.c.
| layer make_rnn_layer | ( | int | batch, |
| int | inputs, | ||
| int | hidden, | ||
| int | outputs, | ||
| int | steps, | ||
| ACTIVATION | activation, | ||
| int | batch_normalize, | ||
| int | log | ||
| ) |
Definition at line 29 of file rnn_layer.c.
| void update_rnn_layer | ( | layer | l, |
| int | batch, | ||
| float | learning_rate, | ||
| float | momentum, | ||
| float | decay | ||
| ) |
Definition at line 76 of file rnn_layer.c.