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)
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 > &)
void	RegisterOperator (const std::string &name, ParserFuncSignature func)
void	RegisterTensorType (const std::string &, ETensorType)
bool	Verbose () const

Private Attributes
std::map< std::string, int >	allInputs
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 145 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 231 of file RModelParser_ONNX.cxx.

GetTensorType()

ETensorType TMVA::Experimental::SOFIE::RModelParser_ONNX::GetTensorType

(

const std::string &

name

)

Definition at line 251 of file RModelParser_ONNX.cxx.

IsRegisteredOperator()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredOperator

(

const std::string &

name

)

Definition at line 226 of file RModelParser_ONNX.cxx.

IsRegisteredTensorType()

bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredTensorType

(

const std::string &

name

)

Definition at line 246 of file RModelParser_ONNX.cxx.

Parse()

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

Definition at line 312 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 355 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 258 of file RModelParser_ONNX.cxx.

RegisterOperator()

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

Definition at line 221 of file RModelParser_ONNX.cxx.

RegisterTensorType()

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

Definition at line 241 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

allInputs

std::map<std::string, int> TMVA::Experimental::SOFIE::RModelParser_ONNX::allInputs

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 35 of file RModelParser_ONNX.hxx.

fTensorTypeMap

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

private

Definition at line 37 of file RModelParser_ONNX.hxx.

fVerbose

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

private

Definition at line 33 of file RModelParser_ONNX.hxx.