LayerTypes.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
#include <type_traits>
 
namespace FlexNN::Layers {
 
enum class LayerType : uint8_t {
  Dense = 0,       
  Conv1D = 1,      
  BatchNorm1D = 2, 
  MaxPool1D = 3,   
  AvgPool1D = 4    
};
static_assert(sizeof(LayerType) == 1, "LayerType must be 1 byte for model.bin");
static_assert(std::is_same_v<std::underlying_type_t<LayerType>, uint8_t>,
              "LayerType underlying type must be uint8_t");
 
enum class DType : uint16_t {
  F32 = 0 
};
static_assert(sizeof(DType) == 2, "DType must be 2 bytes for model.bin");
static_assert(std::is_same_v<std::underlying_type_t<DType>, uint16_t>,
              "DType underlying type must be uint16_t");
 
struct DenseParams {
  int inputSize = 0;  
  int outputSize = 0; 
};
 
struct Conv1DParams {
  int inChannels = 0;
  int outChannels = 0;
  int kernelSize = 0;
  int stride = 1;
  int padding = 0;
  int dilation = 1; 
};
 
struct Pool1DParams {
  int channels = 0;
  int kernelSize = 0;
  int stride = 1;  
  int padding = 0;
};
 
struct BatchNormParams {
  int numFeatures = 0;
  float epsilon = 1e-5f;
  float momentum = 0.1f;
};
 
} // namespace FlexNN::Layers