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< 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)

Private Attributes
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 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.