Functions
gru_layer.c File Reference
#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>
Include dependency graph for gru_layer.c:

Go to the source code of this file.

Functions

void backward_gru_layer (layer l, network_state state)
void forward_gru_layer (layer l, network_state state)
static void increment_layer (layer *l, int steps)
layer make_gru_layer (int batch, int inputs, int outputs, int steps, int batch_normalize)
void update_gru_layer (layer l, int batch, float learning_rate, float momentum, float decay)

Function Documentation

void backward_gru_layer ( layer  l,
network_state  state 
)

Definition at line 194 of file gru_layer.c.

void forward_gru_layer ( layer  l,
network_state  state 
)

Definition at line 118 of file gru_layer.c.

static void increment_layer ( layer l,
int  steps 
) [static]

Definition at line 13 of file gru_layer.c.

layer make_gru_layer ( int  batch,
int  inputs,
int  outputs,
int  steps,
int  batch_normalize 
)

Definition at line 29 of file gru_layer.c.

void update_gru_layer ( layer  l,
int  batch,
float  learning_rate,
float  momentum,
float  decay 
)

Definition at line 111 of file gru_layer.c.



rail_object_detector
Author(s):
autogenerated on Sat Jun 8 2019 20:26:31