METNetSigHandler.h File Reference

#include <string>
#include <onnxruntime_cxx_api.h>
#include <ColumnarMode/ColumnarMode.h>

Include dependency graph for METNetSigHandler.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 METNetSigHandler.h.

                         {
 
  class METNetSigHandler {
 
  public:
 
    // Constructor with parameters
    METNetSigHandler(const std::string& modelName);
 
    // Destructor
    virtual ~METNetSigHandler();
 
    // Public methods
    int initialize();
    int getReqSize() const;
    std::vector<float> predict(std::vector<float> inputs) const;
 
  private:
 
    // Default constructor
    METNetSigHandler();
 
    // 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;
    std::unique_ptr<Ort::Session> m_onnxSession;
  };
 
}