OnnxRuntimeSvc.h

Go to the documentation of this file.

// Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
#ifndef ATHONNX_ONNXRUNTIMESVC_H
#define ATHONNX_ONNXRUNTIMESVC_H
 
// Local include(s).
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h"
 
// Framework include(s).
#include <AsgServices/AsgService.h>
 
// ONNX include(s).
#include <onnxruntime_cxx_api.h>
 
// System include(s).
#include <memory>
 
namespace AthOnnx {
 
   class OnnxRuntimeSvc : public asg::AsgService, virtual public IOnnxRuntimeSvc {
 
   public:
 
      OnnxRuntimeSvc (const std::string& name, ISvcLocator* svc);
 
      virtual StatusCode initialize() override;
      virtual StatusCode finalize() override;
 
 
 
      virtual Ort::Env& env() const override;
 
 
   private:
      std::unique_ptr< Ort::Env > m_env;
 
   }; // class OnnxRuntimeSvc
 
} // namespace AthOnnx
 
#endif // ATHONNX_ONNXRUNTIMESVC_H