LocalNeuralNetImp.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef LOCALNEURALNETIMP_HPP_INCLUDE
#define LOCALNEURALNETIMP_HPP_INCLUDE
 
#include "LocalNeuralNet.hpp"
 
namespace geopm
{
    class LocalNeuralNetImp : public LocalNeuralNet
    {
        public:
            LocalNeuralNetImp(const std::vector<std::shared_ptr<DenseLayer> > layers);
            TensorOneD forward(const TensorOneD &inp) const override;
            size_t get_input_dim() const override;
            size_t get_output_dim() const override;
 
        private:
            std::vector<std::shared_ptr<DenseLayer> > m_layers;
    };
}
 
#endif  /* LOCALNEURALNETIMP_HPP_INCLUDE */