METNetHandler.h File Reference

#include <string>
#include <mutex>
#include <onnxruntime_cxx_api.h>
#include "CxxUtils/checker_macros.h"
#include <ColumnarMode/ColumnarMode.h>

Include dependency graph for METNetHandler.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
namespace	ADD_CMODE (met)

Function Documentation

ADD_CMODE()

namespace ADD_CMODE

(

met

)

Definition at line 1 of file METNetHandler.h.

                         {
 
  class METNetHandler {
 
  public:
 
    // Constructor with parameters
    METNetHandler(const std::string& modelName);
 
    // Destructor
    virtual ~METNetHandler() = default;
 
    // Public methods
    int initialize();
    unsigned int getReqSize() const;
    void predict(std::vector<float>& outputs, std::vector<float>& inputs) const;
 
  private:
 
    // Default constructor
    METNetHandler() = delete;
 
    // Class properties
    std::string m_modelName; // Path to the onnx file
    std::string m_modelPath; // Output of the path resolver
 
    // Features of the network structure
    size_t m_numInputs;
    size_t m_numOutputs;
    std::vector<int64_t> m_inputDims;
    std::vector<int64_t> m_outputDims;
    std::vector<const char *> m_graphInputNames;
    std::vector<const char *> m_graphOutputNames;
 
    // ONNX session objects
    Ort::Env m_onnxEnv;
    Ort::SessionOptions m_onnxSessionOptions;
    Ort::AllocatorWithDefaultOptions m_onnxAllocator;
    mutable std::unique_ptr<Ort::Session> m_onnxSession ATLAS_THREAD_SAFE {nullptr};
    mutable std::mutex m_onnxMutex ATLAS_THREAD_SAFE;
  };
 
}