Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::RModelParser_ONNX Class Reference

Definition at line 28 of file RModelParser_ONNX.hxx.

Classes

struct  OperatorsMapImpl
 

Public Member Functions

 RModelParser_ONNX () noexcept
 
 ~RModelParser_ONNX ()
 
void CheckGraph (const onnx::GraphProto &g, int &level, std::map< std::string, int > &missingOperators)
 
bool CheckModel (std::string filename, bool verbose=false)
 
std::vector< std::string > GetRegisteredOperators ()
 
ETensorType GetTensorType (const std::string &name)
 
bool IsRegisteredOperator (const std::string &name)
 
bool IsRegisteredTensorType (const std::string &)
 
std::unique_ptr< onnx::ModelProto > LoadModel (std::string filename)
 
RModel Parse (std::string filename, bool verbose=false)
 
void ParseONNXGraph (RModel &model, const onnx::GraphProto &g, std::string name="")
 
std::unique_ptr< ROperatorParseOperator (const size_t, const onnx::GraphProto &, const std::vector< size_t > &, const std::vector< int > &)
 
void RegisterOperator (const std::string &name, ParserFuncSignature func)
 
void RegisterTensorType (const std::string &, ETensorType)
 
bool Verbose () const
 

Private Attributes

std::vector< boolfFusedOperators
 
std::unique_ptr< OperatorsMapImplfOperatorsMapImpl
 
std::unordered_map< std::string, ETensorTypefTensorTypeMap
 
bool fVerbose = false
 

#include <TMVA/RModelParser_ONNX.hxx>

Constructor & Destructor Documentation

◆ RModelParser_ONNX()

TMVA::Experimental::SOFIE::RModelParser_ONNX::RModelParser_ONNX ( )
noexcept

Definition at line 155 of file RModelParser_ONNX.cxx.

◆ ~RModelParser_ONNX()

TMVA::Experimental::SOFIE::RModelParser_ONNX::~RModelParser_ONNX ( )
default

Member Function Documentation

◆ CheckGraph()

void TMVA::Experimental::SOFIE::RModelParser_ONNX::CheckGraph ( const onnx::GraphProto & g,
int & level,
std::map< std::string, int > & missingOperators )

Definition at line 398 of file RModelParser_ONNX.cxx.

◆ CheckModel()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::CheckModel ( std::string filename,
bool verbose = false )

Definition at line 429 of file RModelParser_ONNX.cxx.

◆ GetRegisteredOperators()

std::vector< std::string > TMVA::Experimental::SOFIE::RModelParser_ONNX::GetRegisteredOperators ( )

Definition at line 252 of file RModelParser_ONNX.cxx.

◆ GetTensorType()

ETensorType TMVA::Experimental::SOFIE::RModelParser_ONNX::GetTensorType ( const std::string & name)

Definition at line 274 of file RModelParser_ONNX.cxx.

◆ IsRegisteredOperator()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredOperator ( const std::string & name)

Definition at line 247 of file RModelParser_ONNX.cxx.

◆ IsRegisteredTensorType()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredTensorType ( const std::string & name)

Definition at line 269 of file RModelParser_ONNX.cxx.

◆ LoadModel()

std::unique_ptr< onnx::ModelProto > TMVA::Experimental::SOFIE::RModelParser_ONNX::LoadModel ( std::string filename)

Definition at line 378 of file RModelParser_ONNX.cxx.

◆ Parse()

RModel TMVA::Experimental::SOFIE::RModelParser_ONNX::Parse ( std::string filename,
bool verbose = false )

Definition at line 345 of file RModelParser_ONNX.cxx.

◆ ParseONNXGraph()

void TMVA::Experimental::SOFIE::RModelParser_ONNX::ParseONNXGraph ( RModel & model,
const onnx::GraphProto & g,
std::string name = "" )

Definition at line 455 of file RModelParser_ONNX.cxx.

◆ ParseOperator()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::RModelParser_ONNX::ParseOperator ( const size_t i,
const onnx::GraphProto & graphproto,
const std::vector< size_t > & nodes,
const std::vector< int > & children )

Definition at line 281 of file RModelParser_ONNX.cxx.

◆ RegisterOperator()

void TMVA::Experimental::SOFIE::RModelParser_ONNX::RegisterOperator ( const std::string & name,
ParserFuncSignature func )

Definition at line 242 of file RModelParser_ONNX.cxx.

◆ RegisterTensorType()

void TMVA::Experimental::SOFIE::RModelParser_ONNX::RegisterTensorType ( const std::string & name,
ETensorType type )

Definition at line 264 of file RModelParser_ONNX.cxx.

◆ Verbose()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::Verbose ( ) const
inline

Definition at line 60 of file RModelParser_ONNX.hxx.

Member Data Documentation

◆ fFusedOperators

std::vector<bool> TMVA::Experimental::SOFIE::RModelParser_ONNX::fFusedOperators
private

Definition at line 40 of file RModelParser_ONNX.hxx.

◆ fOperatorsMapImpl

std::unique_ptr<OperatorsMapImpl> TMVA::Experimental::SOFIE::RModelParser_ONNX::fOperatorsMapImpl
private

Definition at line 36 of file RModelParser_ONNX.hxx.

◆ fTensorTypeMap

std::unordered_map<std::string, ETensorType> TMVA::Experimental::SOFIE::RModelParser_ONNX::fTensorTypeMap
private

Definition at line 38 of file RModelParser_ONNX.hxx.

◆ fVerbose

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::fVerbose = false
private

Definition at line 34 of file RModelParser_ONNX.hxx.

  • tmva/sofie_parsers/inc/TMVA/RModelParser_ONNX.hxx
  • tmva/sofie_parsers/src/RModelParser_ONNX.cxx