#include "local_layer.h"
#include "utils.h"
#include "im2col.h"
#include "col2im.h"
#include "blas.h"
#include "gemm.h"
#include <stdio.h>
#include <time.h>
Go to the source code of this file.
Functions | |
void | backward_local_layer (local_layer l, network_state state) |
void | forward_local_layer (const local_layer l, network_state state) |
int | local_out_height (local_layer l) |
int | local_out_width (local_layer l) |
local_layer | make_local_layer (int batch, int h, int w, int c, int n, int size, int stride, int pad, ACTIVATION activation) |
void | update_local_layer (local_layer l, int batch, float learning_rate, float momentum, float decay) |
void backward_local_layer | ( | local_layer | l, |
network_state | state | ||
) |
Definition at line 122 of file local_layer.c.
void forward_local_layer | ( | const local_layer | l, |
network_state | state | ||
) |
Definition at line 91 of file local_layer.c.
int local_out_height | ( | local_layer | l | ) |
Definition at line 10 of file local_layer.c.
int local_out_width | ( | local_layer | l | ) |
Definition at line 18 of file local_layer.c.
local_layer make_local_layer | ( | int | batch, |
int | h, | ||
int | w, | ||
int | c, | ||
int | n, | ||
int | size, | ||
int | stride, | ||
int | pad, | ||
ACTIVATION | activation | ||
) |
Definition at line 26 of file local_layer.c.
void update_local_layer | ( | local_layer | l, |
int | batch, | ||
float | learning_rate, | ||
float | momentum, | ||
float | decay | ||
) |
Definition at line 167 of file local_layer.c.