conv_layer_struct

This structure contains all parameters necessary for Convolutional layers.

#include <layer.h>

Public Attributes

uint32_t CO;

Number of output channels

uint32_t CI;

Number of input channels

uint32_t IH;

Height of input feature map

uint32_t IW;

Width of input feature map

uint32_t OH;

Height of output feature map

uint32_t OW;

Width of output feature map

uint32_t FH;

Height of filter

uint32_t FW;

Width of filter

uint32_t pad;

Padding on all sides

double * ifmap;

Pointer to input feature map

double * weights;

Pointer to weights

double * ofmap;

Pointer to output feature map

uint32_t TILE_CI;

Tiling factor of input channel

uint32_t cluster2cluster;

Flag for enabling cluster 2 cluster communication

uint32_t im2col;

Flag for enabling im2col + GEMM

double * gamma;

Pointer to gamma for BatchNorm

double * beta;

Pointer to beta for BatchNorm

precision_t dtype;

Updated on 2023-06-19 at 09:43:56 +0000