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< ROperator >	ParseOperator (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< bool >	fFusedOperators
std::unique_ptr< OperatorsMapImpl >	fOperatorsMapImpl
std::unordered_map< std::string, ETensorType >	fTensorTypeMap
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.