batch_norm_fold.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "pulsatrix/batch_norm_module.hpp"
#include "pulsatrix/conv2d_module.hpp"
 
namespace pulsatrix {
 
class BatchNormFold {
public:
    BatchNormFold(Conv2DModule& conv, BatchNormModule& bn);
    ~BatchNormFold();
 
    BatchNormFold(const BatchNormFold&) = delete;
    BatchNormFold& operator=(const BatchNormFold&) = delete;
    BatchNormFold(BatchNormFold&&) = delete;
    BatchNormFold& operator=(BatchNormFold&&) = delete;
 
private:
    Conv2DModule& conv_;
    BatchNormModule& bn_;
    std::vector<float> original_kernel_;
    std::vector<float> original_bias_;
};
 
}  // namespace pulsatrix