File stub_backend.hpp

File List > common > src > inference > stub_backend.hpp

Go to the documentation of this file

#ifndef STUB_BACKEND_HPP
#define STUB_BACKEND_HPP

#include "inference_backend.hpp"

namespace emulator {
namespace inference {

class StubBackend : public InferenceBackend {
public:
  StubBackend() = default;
  ~StubBackend() override = default;

  bool initialize(const InferenceConfig &config) override;

  bool infer(const double *inputs, double *outputs, int batch_size) override;

  void finalize() override;

  std::string name() const override { return "Stub"; }

  bool is_initialized() const override { return m_initialized; }

  BackendType type() const override { return BackendType::STUB; }

private:
  bool m_initialized = false; 
  InferenceConfig m_config;   
};

} // namespace inference
} // namespace emulator

#endif // STUB_BACKEND_HPP