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

Definition at line 27 of file RModelParser_ONNX.hxx.

Classes

struct  OperatorsMapImpl
 

Public Member Functions

 RModelParser_ONNX () noexcept
 
 ~RModelParser_ONNX ()
 
std::vector< std::string > GetRegisteredOperators ()
 
ETensorType GetTensorType (const std::string &name)
 
bool IsRegisteredOperator (const std::string &name)
 
bool IsRegisteredTensorType (const std::string &)
 
RModel Parse (std::string filename, bool verbose=false)
 
std::unique_ptr< ROperatorParseOperator (const size_t, const onnx::GraphProto &, const std::vector< size_t > &)
 
void RegisterOperator (const std::string &name, ParserFuncSignature func)
 
void RegisterTensorType (const std::string &, ETensorType)
 

Private Attributes

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 75 of file RModelParser_ONNX.cxx.

◆ ~RModelParser_ONNX()

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

Member Function Documentation

◆ GetRegisteredOperators()

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

Definition at line 144 of file RModelParser_ONNX.cxx.

◆ GetTensorType()

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

Definition at line 164 of file RModelParser_ONNX.cxx.

◆ IsRegisteredOperator()

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

Definition at line 139 of file RModelParser_ONNX.cxx.

◆ IsRegisteredTensorType()

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

Definition at line 159 of file RModelParser_ONNX.cxx.

◆ Parse()

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

Definition at line 218 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 
)

Definition at line 171 of file RModelParser_ONNX.cxx.

◆ RegisterOperator()

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

Definition at line 134 of file RModelParser_ONNX.cxx.

◆ RegisterTensorType()

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

Definition at line 154 of file RModelParser_ONNX.cxx.

Member Data Documentation

◆ fOperatorsMapImpl

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

Definition at line 34 of file RModelParser_ONNX.hxx.

◆ fTensorTypeMap

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

Definition at line 36 of file RModelParser_ONNX.hxx.

◆ fVerbose

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

Definition at line 32 of file RModelParser_ONNX.hxx.

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