o2::ml::OrtModel Class Reference

#include <OrtInterface.h>

Classes
struct	OrtVariables

Public Member Functions
	OrtModel ()=default
	OrtModel (std::unordered_map< std::string, std::string > optionsMap)
void	init (std::unordered_map< std::string, std::string > optionsMap)
void	reset (std::unordered_map< std::string, std::string >)
virtual	~OrtModel ()=default
template<class I , class O >
std::vector< O >	v2v (std::vector< I > &, bool=true)
template<class I , class O >
std::vector< O >	inference (std::vector< I > &)
template<class I , class O >
std::vector< O >	inference (std::vector< std::vector< I > > &)
void	resetSession ()
std::vector< std::vector< int64_t > >	getNumInputNodes () const
std::vector< std::vector< int64_t > >	getNumOutputNodes () const
std::vector< std::string >	getInputNames () const
std::vector< std::string >	getOutputNames () const
void	setActiveThreads (int threads)
template<>
std::vector< float >	inference (std::vector< float > &input)

Detailed Description

Definition at line 35 of file OrtInterface.h.

Constructor & Destructor Documentation

OrtModel() [1/2]

o2::ml::OrtModel::OrtModel ( )

default

OrtModel() [2/2]

o2::ml::OrtModel::OrtModel ( std::unordered_map< std::string, std::string >  optionsMap )

inline

Definition at line 41 of file OrtInterface.h.

~OrtModel()

virtual o2::ml::OrtModel::~OrtModel ( )

virtualdefault

Member Function Documentation

getInputNames()

std::vector< std::string > o2::ml::OrtModel::getInputNames ( ) const

inline

Definition at line 66 of file OrtInterface.h.

getNumInputNodes()

std::vector< std::vector< int64_t > > o2::ml::OrtModel::getNumInputNodes ( ) const

inline

Definition at line 64 of file OrtInterface.h.

getNumOutputNodes()

std::vector< std::vector< int64_t > > o2::ml::OrtModel::getNumOutputNodes ( ) const

inline

Definition at line 65 of file OrtInterface.h.

getOutputNames()

std::vector< std::string > o2::ml::OrtModel::getOutputNames ( ) const

inline

Definition at line 67 of file OrtInterface.h.

inference() [1/3]

template<>

std::vector< float > o2::ml::OrtModel::inference

(

std::vector< float > &

input

)

Definition at line 220 of file OrtInterface.cxx.

inference() [2/3]

template<class I , class O >

std::vector< O > o2::ml::OrtModel::inference

(

std::vector< I > &

input

)

Definition at line 191 of file OrtInterface.cxx.

inference() [3/3]

template<class I , class O >

std::vector< O > o2::ml::OrtModel::inference

(

std::vector< std::vector< I > > &

input

)

Definition at line 205 of file OrtInterface.cxx.

init()

void o2::ml::OrtModel::init ( std::unordered_map< std::string, std::string >  optionsMap )

inline

Definition at line 42 of file OrtInterface.h.

reset()

void o2::ml::OrtModel::reset

(

std::unordered_map< std::string, std::string >

optionsMap

)

Definition at line 38 of file OrtInterface.cxx.

resetSession()

void o2::ml::OrtModel::resetSession

(

)

Definition at line 170 of file OrtInterface.cxx.

setActiveThreads()

void o2::ml::OrtModel::setActiveThreads ( int  threads )

inline

Definition at line 69 of file OrtInterface.h.

v2v()

template<class I , class O >

std::vector< O > o2::ml::OrtModel::v2v	(	std::vector< I > &	input,
		bool	clearInput = true
	)

Definition at line 176 of file OrtInterface.cxx.

---

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

• /home/runner/work/AliceO2/AliceO2/Common/ML/include/ML/OrtInterface.h
• /home/runner/work/AliceO2/AliceO2/Common/ML/src/OrtInterface.cxx