Ringer::onnx::Model Class Reference

#include <Model.h>

Collaboration diagram for Ringer::onnx::Model:

Public Member Functions
	Model (const std::string &modelPath, AthOnnx::IOnnxRuntimeSvc *svc, float etmin, float etmax, float etamin, float etamax, unsigned barcode)
	Constructor. More...
	~Model ()=default
	Destructor. More...
void	compile ()
float	etMin () const
	Get the Et lower edge. More...
float	etMax () const
	Get the Et high edge. More...
float	etaMin () const
	Get the Eta lower edge. More...
float	etaMax () const
	Get the Eta high edge. More...
float	predict (std::vector< std::vector< float > > &) const
	Calculate the disriminant. More...
unsigned	barcode () const

Private Attributes
std::shared_ptr< Ort::Session >	m_session
std::vector< std::vector< int64_t > >	m_input_node_dims
std::vector< int64_t >	m_output_node_dims
std::vector< const char * >	m_input_node_names
std::vector< const char * >	m_output_node_names
float	m_etmin
float	m_etmax
float	m_etamin
float	m_etamax
unsigned	m_barcode

Detailed Description

Definition at line 21 of file Model.h.

Constructor & Destructor Documentation

Model()

Ringer::onnx::Model::Model	(	const std::string &	modelPath,
		AthOnnx::IOnnxRuntimeSvc *	svc,
		float	etmin,
		float	etmax,
		float	etamin,
		float	etamax,
		unsigned	barcode
	)

Constructor.

Definition at line 14 of file Model.cxx.

                                   :
      m_etmin(etmin),
      m_etmax(etmax),
      m_etamin(etamin),
      m_etamax(etamax),
      m_barcode(barcode)
 
    {
      // Some ORT related initialization
      Ort::SessionOptions sessionOptions;
      sessionOptions.SetIntraOpNumThreads( 1 );
      sessionOptions.SetGraphOptimizationLevel( ORT_ENABLE_BASIC );  
      m_session = std::make_shared< Ort::Session >( svc->env(), modelPath.c_str(), sessionOptions );
    }   

~Model()

Ringer::onnx::Model::~Model ( )

default

Destructor.

Member Function Documentation

barcode()

unsigned Ringer::onnx::Model::barcode ( ) const

inline

Definition at line 50 of file Model.h.

{ return m_barcode; };

compile()

void Ringer::onnx::Model::compile

(

)

Definition at line 32 of file Model.cxx.

    {
      Ort::AllocatorWithDefaultOptions allocator;  
      size_t num_input_nodes = m_session->GetInputCount();
      
      for( std::size_t i = 0; i < num_input_nodes; i++ ) {
        char* input_name = m_session->GetInputNameAllocated(i, allocator).release();
        m_input_node_names.push_back(input_name);
        Ort::TypeInfo type_info = m_session->GetInputTypeInfo(i);
        auto tensor_info = type_info.GetTensorTypeAndShapeInfo();
        auto input_node_dims = tensor_info.GetShape();
        for (std::size_t j = 0; j < input_node_dims.size(); j++){
          if(input_node_dims[j]<0)
            input_node_dims[j] =1;
        }  
        m_input_node_dims.push_back(input_node_dims);
      }
      
     
      // Always have only one output
      size_t num_output_nodes = m_session->GetOutputCount();      
      for( std::size_t i = 0; i < num_output_nodes; i++ ) {
        char* output_name = m_session->GetOutputNameAllocated(i, allocator).release();
        m_output_node_names.push_back(output_name);
      }
 
      
    }

etaMax()

float Ringer::onnx::Model::etaMax ( ) const

inline

Get the Eta high edge.

Definition at line 45 of file Model.h.

{ return m_etamax; };

etaMin()

float Ringer::onnx::Model::etaMin ( ) const

inline

Get the Eta lower edge.

Definition at line 42 of file Model.h.

{ return m_etamin; };

etMax()

float Ringer::onnx::Model::etMax ( ) const

inline

Get the Et high edge.

Definition at line 39 of file Model.h.

{ return m_etmax; };

etMin()

float Ringer::onnx::Model::etMin ( ) const

inline

Get the Et lower edge.

Definition at line 36 of file Model.h.

{ return m_etmin; };

predict()

float Ringer::onnx::Model::predict

(

std::vector< std::vector< float > > &

input_vecs

)

const

Calculate the disriminant.

Definition at line 65 of file Model.cxx.

    {
      auto memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
      std::vector<Ort::Value> ort_inputs;
 
      size_t num_inputs = input_vecs.size();
      for( size_t i=0; i < num_inputs; ++i ){
        ort_inputs.emplace_back( Ort::Value::CreateTensor<float>(memory_info, input_vecs[i].data(), 
                                                                  input_vecs[i].size(), m_input_node_dims[i].data(), 
                                                                  m_input_node_dims[i].size()) );
      }
 
      // score model & input tensor, get back output tensor
      auto output_tensors = m_session->Run(Ort::RunOptions{nullptr}, m_input_node_names.data(), ort_inputs.data(), 
                                           ort_inputs.size(), m_output_node_names.data(), 1);
      
      assert(output_tensors.size() == 1 && output_tensors.front().IsTensor());
    
      // Get pointer to output tensor float values
      float* output_arr = output_tensors.front().GetTensorMutableData<float>();
      
      return output_arr[0];
    }

Member Data Documentation

m_barcode

unsigned Ringer::onnx::Model::m_barcode

private

Definition at line 63 of file Model.h.

m_etamax

float Ringer::onnx::Model::m_etamax

private

Definition at line 62 of file Model.h.

m_etamin

float Ringer::onnx::Model::m_etamin

private

Definition at line 62 of file Model.h.

m_etmax

float Ringer::onnx::Model::m_etmax

private

Definition at line 61 of file Model.h.

m_etmin

float Ringer::onnx::Model::m_etmin

private

Definition at line 61 of file Model.h.

m_input_node_dims

std::vector<std::vector<int64_t> > Ringer::onnx::Model::m_input_node_dims

private

Definition at line 56 of file Model.h.

m_input_node_names

std::vector<const char*> Ringer::onnx::Model::m_input_node_names

private

Definition at line 58 of file Model.h.

m_output_node_dims

std::vector<int64_t> Ringer::onnx::Model::m_output_node_dims

private

Definition at line 57 of file Model.h.

m_output_node_names

std::vector<const char*> Ringer::onnx::Model::m_output_node_names

private

Definition at line 59 of file Model.h.

m_session

std::shared_ptr<Ort::Session> Ringer::onnx::Model::m_session

private

Definition at line 55 of file Model.h.

---

The documentation for this class was generated from the following files:

• Model.h
• Model.cxx