data_loader.h

#include "oneapi/dnnl/dnnl.hpp"
#ifndef _DATA_LOADER
#define _DATA_LOADER
 
#include <iostream>
#include <stdexcept>
#include <cmath>
#include <random>
#include <functional>   // std::multiplies
#include <numeric>      // std::accumulate
#include "intel_utils.h"
 
class DataLoader{
    public:
        int dataset_size; 
        int minibatch_size; 
        std::vector<float> dataset; 
        std::vector<float> dataset_labels; 
        std::vector<float> curr_batch; 
        std::vector<float> curr_batch_labels; 
        void write_to_memory(dnnl::memory dst_mem_features, dnnl::memory dst_mem_labels);  
        DataLoader(std::string features_path, std::string labels_path, int _minibatch_size, std::vector<int> dataset_shape, dnnl::engine _eng);
    private:
        int dataset_idx, sample_size;
        void load_from_file(std::string filename, std::vector<float> &data); 
        dnnl::engine eng;
    
};
 
#endif