![]() |
darknet
v3
|
#include "gru_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 | |
layer | make_gru_layer (int batch, int inputs, int outputs, int steps, int batch_normalize, int adam) |
void | update_gru_layer (layer l, update_args a) |
void | forward_gru_layer (layer l, network net) |
void | backward_gru_layer (layer l, network net) |
Definition at line 204 of file gru_layer.c.
Definition at line 128 of file gru_layer.c.
layer make_gru_layer | ( | int | batch, |
int | inputs, | ||
int | outputs, | ||
int | steps, | ||
int | batch_normalize, | ||
int | adam | ||
) |
Definition at line 29 of file gru_layer.c.
void update_gru_layer | ( | layer | l, |
update_args | a | ||
) |
Definition at line 118 of file gru_layer.c.