ConvLayers.h

//
//  ConvLayers.h
//  ConvNet
//
//  Created by Márton Szemenyei on 2017. 09. 28..
//  Copyright © 2017. Márton Szemenyei. All rights reserved.
//

#ifndef ConvLayers_h
#define ConvLayers_h

#include "Layer.h"
#include "Utils.h"

class ConvLayer : public Layer
{
private:
    float *weights;
    float *bias;
    Tuple size;
    Tuple stride;
    Tuple padding;
    Tuple dilation;
    bool hasBias;
    int32_t *lut;
    
public:
    ConvLayer(int32_t _h, int32_t _w, int32_t _inCh, int32_t _outCh, Tuple _size, Tuple _stride, Tuple _padding, Tuple _dilation, ACTIVATION _activation, bool _hasBias);
    ~ConvLayer();
    void forward();
    bool HasBias();
    int32_t getWorkSpaceSize();
    bool loadWeights( std::ifstream &file );
    void print();
};

class TransposedConvLayer : public Layer
{
private:
    float *weights;
    float *bias;
    Tuple size;
    Tuple stride;
    Tuple padding;
    int32_t outPadding;
    bool hasBias;
public:
    TransposedConvLayer(int32_t _h, int32_t _w, int32_t _inCh, int32_t _outCh, Tuple _size, Tuple _stride, Tuple _padding, int32_t _outPadding, ACTIVATION _activation, bool _hasBias);
    ~TransposedConvLayer();
    void forward();
    bool HasBias();
    int32_t getWorkSpaceSize();
    bool loadWeights( std::ifstream &file );
    void print();
};

class FCLayer : public Layer
{
private:
    float *weights;
    float *bias;
    bool hasBias;
public:
    FCLayer(int32_t _inCh, int32_t _outCh, ACTIVATION _activation, bool _hasBias);
    FCLayer();
    void forward();
    bool HasBias();
    int32_t getWorkSpaceSize();
    bool loadWeights( std::ifstream &file );
    void print();
};

#endif /* ConvLayers_h */