Project
Loading...
Searching...
No Matches
Classes | Public Member Functions | List of all members
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 >)
 
bool isInitialized ()
 
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 > > &)
 
template<class I , class O >
void inference (I *, size_t, O *)
 
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)
 

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 70 of file OrtInterface.h.

◆ getNumInputNodes()

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

Definition at line 68 of file OrtInterface.h.

◆ getNumOutputNodes()

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

Definition at line 69 of file OrtInterface.h.

◆ getOutputNames()

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

Definition at line 71 of file OrtInterface.h.

◆ inference() [1/3]

template<class I , class O >
void o2::ml::OrtModel::inference ( I *  input,
size_t  input_size,
O *  output 
)

Definition at line 227 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 203 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 249 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.

◆ isInitialized()

bool o2::ml::OrtModel::isInitialized ( )
inline

Definition at line 44 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 172 of file OrtInterface.cxx.

◆ setActiveThreads()

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

Definition at line 73 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 178 of file OrtInterface.cxx.

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