Namespaces
namespace	BLAS

namespace	PyKeras

namespace	PyTorch

namespace	UTILITY

Classes
struct	BinaryOperatorTrait

struct	BinaryOperatorTrait< T, Add >

struct	BinaryOperatorTrait< T, Div >

struct	BinaryOperatorTrait< T, Mul >

struct	BinaryOperatorTrait< T, Pow >

struct	BinaryOperatorTrait< T, Sub >

struct	Dim

struct	GNN_Data

struct	GNN_Init

struct	GraphIndependent_Init

struct	InitializedTensor

struct	InputTensorInfo

struct	NaryOperatorTraits

struct	NaryOperatorTraits< float, EBasicNaryOperator::Mean >

struct	NaryOperatorTraits< T, EBasicNaryOperator::Max >

struct	NaryOperatorTraits< T, EBasicNaryOperator::Mean >

struct	NaryOperatorTraits< T, EBasicNaryOperator::Min >

struct	NaryOperatorTraits< T, EBasicNaryOperator::Sum >

struct	RAttributes_Pool

class	RFunction

class	RFunction_Aggregate

class	RFunction_Mean

class	RFunction_MLP

class	RFunction_Sum

class	RFunction_Update

class	RModel

class	RModel_Base

class	RModel_GNN

class	RModel_GNNBase

class	RModel_GraphIndependent

class	RModelParser_ONNX

class	ROperator

class	ROperator_BasicBinary

class	ROperator_BasicNary

class	ROperator_BasicUnary

class	ROperator_BatchNormalization

class	ROperator_Cast

class	ROperator_Concat

class	ROperator_Conv

class	ROperator_ConvTranspose
	Transposed Convolution operator. More...

class	ROperator_Custom

class	ROperator_Erf

class	ROperator_Expand

class	ROperator_Gather

class	ROperator_Gemm

class	ROperator_GRU
	Gated Recurrent Unit operator. More...

class	ROperator_Identity

class	ROperator_LayerNormalization

class	ROperator_LeakyRelu

class	ROperator_LSTM
	Long Short-Term Memory operator. More...

class	ROperator_Pool

class	ROperator_Reduce

class	ROperator_Relu

class	ROperator_Reshape

class	ROperator_RNN
	Recurrent Neural Network operator. More...

class	ROperator_Selu

class	ROperator_Shape

class	ROperator_Sigmoid

class	ROperator_Slice

class	ROperator_Softmax

class	ROperator_Swish

class	ROperator_Tanh

class	ROperator_Transpose

struct	TensorInfo

struct	UnaryOpTraits

struct	UnaryOpTraits< T, EBasicUnaryOperator::kExp >

struct	UnaryOpTraits< T, EBasicUnaryOperator::kLog >

struct	UnaryOpTraits< T, EBasicUnaryOperator::kNeg >

struct	UnaryOpTraits< T, EBasicUnaryOperator::kReciprocal >

struct	UnaryOpTraits< T, EBasicUnaryOperator::kSqrt >

Typedefs
typedef std::int64_t	int_t

using	ParserFuncSignature = std::function< std::unique_ptr< ROperator >(RModelParser_ONNX &, const onnx::NodeProto &)>

using	ParserFuseFuncSignature = std::function< std::unique_ptr< ROperator >(RModelParser_ONNX &, const onnx::NodeProto &, const onnx::NodeProto &)>

Enumerations
enum class	Activation { RELU = 0x0 , Invalid = 0x1 }

enum	EBasicBinaryOperator { Add , Sub , Mul , Div , Pow }

enum class	EBasicNaryOperator { Max , Min , Mean , Sum }

enum class	EBasicUnaryOperator { kReciprocal , kSqrt , kNeg , kExp , kLog }

enum	EReduceOpMode { ReduceMean , ReduceSum , ReduceSumsquare , ReduceProd , InvalidReduceOp }

enum class	ETensorType { UNDEFINED = 0 , FLOAT = 1 , UNINT8 = 2 , INT8 = 3 , UINT16 = 4 , INT16 = 5 , INT32 = 6 , INT64 = 7 , STRING = 8 , BOOL = 9 , FLOAT16 = 10 , DOUBLE = 11 , UINT32 = 12 , UINT64 = 13 , COMPLEX64 = 14 , COMPLEX28 = 15 , BFLOAT16 = 16 }

enum class	FunctionReducer { INVALID =0 , SUM =1 , MEAN =2 }

enum class	FunctionRelation { INVALID =0 , NODES_EDGES =1 , NODES_GLOBALS =2 , EDGES_GLOBALS =3 }

enum class	FunctionTarget { INVALID =0 , NODES =1 , EDGES =2 , GLOBALS =3 }

enum class	FunctionType { UPDATE =0 , AGGREGATE =1 }

enum class	GraphType { INVALID =0 , GNN =1 , GraphIndependent =2 }

enum class	Options { kDefault = 0x0 , kNoSession = 0x1 , kNoWeightFile = 0x2 , kRootBinaryWeightFile = 0x4 , kGNN = 0x8 , kGNNComponent = 0x10 }

enum	PoolOpMode { InvalidPool , MaxPool , AveragePool , GlobalAveragePool }

enum	ReshapeOpMode { Reshape , Flatten , Squeeze , Unsqueeze }

enum class	WeightFileType { None , RootBinary , Text }

Functions
GNN_Data	Concatenate (GNN_Data &data1, GNN_Data &data2, int axis=0)

template<typename T >
TMVA::Experimental::RTensor< T >	Concatenate (TMVA::Experimental::RTensor< T > &t1, TMVA::Experimental::RTensor< T > &t2, int axis=0)

std::vector< Dim >	ConvertShapeToDim (std::vector< size_t > shape)

std::size_t	ConvertShapeToLength (std::vector< size_t > shape)

std::string	ConvertShapeToString (std::vector< size_t > shape)

ETensorType	ConvertStringToType (std::string type)

std::string	ConvertTypeToString (ETensorType type)

GNN_Data	Copy (const GNN_Data &data)

template<typename T >
ETensorType	GetTemplatedType (T)

std::underlying_type_t< Options >	operator\| (Options opA, Options opB)

std::underlying_type_t< Options >	operator\| (std::underlying_type_t< Options > opA, Options opB)

template<EBasicBinaryOperator Op>
std::unique_ptr< ROperator >	ParseBasicBinary (RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto)

template<EBasicNaryOperator Op>
std::unique_ptr< ROperator >	ParseBasicNary (RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto)

template<EBasicUnaryOperator Op>
std::unique_ptr< ROperator >	ParseBasicUnary (RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto)

template<EReduceOpMode Op>
std::unique_ptr< ROperator >	ParseReduce (RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto)

Variables
ParserFuncSignature	ParseAdd

ParserFuncSignature	ParseBatchNormalization

ParserFuncSignature	ParseCast

ParserFuncSignature	ParseConcat

ParserFuncSignature	ParseConv

ParserFuncSignature	ParseConvTranspose

ParserFuncSignature	ParseDiv

ParserFuncSignature	ParseErf

ParserFuncSignature	ParseExp

ParserFuncSignature	ParseExpand

ParserFuseFuncSignature	ParseFuseConvAdd

ParserFuseFuncSignature	ParseFuseConvTransposeAdd

ParserFuseFuncSignature	ParseFuseMatMulAdd

ParserFuncSignature	ParseGather

ParserFuncSignature	ParseGemm

ParserFuncSignature	ParseGRU

ParserFuncSignature	ParseIdentity

ParserFuncSignature	ParseLayerNormalization

ParserFuncSignature	ParseLeakyRelu

ParserFuncSignature	ParseLog

ParserFuncSignature	ParseLSTM

ParserFuncSignature	ParseMatMul

ParserFuncSignature	ParseMax

ParserFuncSignature	ParseMean

ParserFuncSignature	ParseMin

ParserFuncSignature	ParseMul

ParserFuncSignature	ParseNeg

ParserFuncSignature	ParsePool

ParserFuncSignature	ParsePow

ParserFuncSignature	ParseReciprocal

ParserFuncSignature	ParseReduceMean

ParserFuncSignature	ParseReduceProd

ParserFuncSignature	ParseReduceSumsquare

ParserFuncSignature	ParseRelu

ParserFuncSignature	ParseReshape

ParserFuncSignature	ParseRNN

ParserFuncSignature	ParseSelu

ParserFuncSignature	ParseShape

ParserFuncSignature	ParseSigmoid

ParserFuncSignature	ParseSlice

ParserFuncSignature	ParseSoftmax

ParserFuncSignature	ParseSqrt

ParserFuncSignature	ParseSub

ParserFuncSignature	ParseSum

ParserFuncSignature	ParseTanh

ParserFuncSignature	ParseTranspose

Typedef Documentation

◆ int_t

typedef std::int64_t TMVA::Experimental::SOFIE::int_t

Definition at line 28 of file SOFIE_common.hxx.

◆ ParserFuncSignature

using TMVA::Experimental::SOFIE::ParserFuncSignature = typedef std::function<std::unique_ptr<ROperator>(RModelParser_ONNX & , const onnx::NodeProto & )>

Definition at line 22 of file RModelParser_ONNX.hxx.

◆ ParserFuseFuncSignature

using TMVA::Experimental::SOFIE::ParserFuseFuncSignature = typedef std::function<std::unique_ptr<ROperator> (RModelParser_ONNX& , const onnx::NodeProto& , const onnx::NodeProto& )>

Definition at line 24 of file RModelParser_ONNX.hxx.

Enumeration Type Documentation

◆ Activation

enum class TMVA::Experimental::SOFIE::Activation

strong

Enumerator
RELU
Invalid

Definition at line 13 of file RFunction_MLP.hxx.

◆ EBasicBinaryOperator

enum TMVA::Experimental::SOFIE::EBasicBinaryOperator

Enumerator
Add
Sub
Mul
Div
Pow

Definition at line 14 of file ROperator_BasicBinary.hxx.

◆ EBasicNaryOperator

enum class TMVA::Experimental::SOFIE::EBasicNaryOperator

strong

Enumerator
Max
Min
Mean
Sum

Definition at line 16 of file ROperator_BasicNary.hxx.

◆ EBasicUnaryOperator

enum class TMVA::Experimental::SOFIE::EBasicUnaryOperator

strong

Enumerator
kReciprocal
kSqrt
kNeg
kExp
kLog

Definition at line 12 of file ROperator_BasicUnary.hxx.

◆ EReduceOpMode

enum TMVA::Experimental::SOFIE::EReduceOpMode

Enumerator
ReduceMean
ReduceSum
ReduceSumsquare
ReduceProd
InvalidReduceOp

Definition at line 19 of file ROperator_Reduce.hxx.

◆ ETensorType

enum class TMVA::Experimental::SOFIE::ETensorType

strong

Enumerator
UNDEFINED
FLOAT
UNINT8
INT8
UINT16
INT16
INT32
INT64
STRING
BOOL
FLOAT16
DOUBLE
UINT32
UINT64
COMPLEX64
COMPLEX28
BFLOAT16

Definition at line 23 of file SOFIE_common.hxx.

◆ FunctionReducer

enum class TMVA::Experimental::SOFIE::FunctionReducer

strong

Enumerator
INVALID
SUM
MEAN

Definition at line 108 of file RModel_Base.hxx.

◆ FunctionRelation

enum class TMVA::Experimental::SOFIE::FunctionRelation

strong

Enumerator
INVALID
NODES_EDGES
NODES_GLOBALS
EDGES_GLOBALS

Definition at line 111 of file RModel_Base.hxx.

◆ FunctionTarget

enum class TMVA::Experimental::SOFIE::FunctionTarget

strong

Enumerator
INVALID
NODES
EDGES
GLOBALS

Definition at line 105 of file RModel_Base.hxx.

◆ FunctionType

enum class TMVA::Experimental::SOFIE::FunctionType

strong

Enumerator
UPDATE
AGGREGATE

Definition at line 102 of file RModel_Base.hxx.

◆ GraphType

enum class TMVA::Experimental::SOFIE::GraphType

strong

Enumerator
INVALID
GNN
GraphIndependent

Definition at line 98 of file RModel_Base.hxx.

◆ Options

enum class TMVA::Experimental::SOFIE::Options

strong

Enumerator
kDefault
kNoSession
kNoWeightFile
kRootBinaryWeightFile
kGNN
kGNNComponent

Definition at line 22 of file RModel_Base.hxx.

◆ PoolOpMode

enum TMVA::Experimental::SOFIE::PoolOpMode

Enumerator
InvalidPool
MaxPool
AveragePool
GlobalAveragePool

Definition at line 31 of file ROperator_Pool.hxx.

◆ ReshapeOpMode

enum TMVA::Experimental::SOFIE::ReshapeOpMode

Enumerator
Reshape
Flatten
Squeeze
Unsqueeze

Definition at line 15 of file ROperator_Reshape.hxx.

◆ WeightFileType

enum class TMVA::Experimental::SOFIE::WeightFileType

strong

Enumerator
None
RootBinary
Text

Definition at line 31 of file RModel_Base.hxx.

Function Documentation

◆ Concatenate() [1/2]

GNN_Data TMVA::Experimental::SOFIE::Concatenate	(	GNN_Data &	data1,
		GNN_Data &	data2,
		int	axis = `0`
	)

inline

Definition at line 473 of file SOFIE_common.hxx.

◆ Concatenate() [2/2]

template<typename T >

TMVA::Experimental::RTensor< T > TMVA::Experimental::SOFIE::Concatenate	(	TMVA::Experimental::RTensor< T > &	t1,
		TMVA::Experimental::RTensor< T > &	t2,
		int	axis = `0`
	)

Definition at line 440 of file SOFIE_common.hxx.

◆ ConvertShapeToDim()

std::vector< Dim > TMVA::Experimental::SOFIE::ConvertShapeToDim ( std::vector< size_t > shape )

Definition at line 10 of file SOFIE_common.cxx.

◆ ConvertShapeToLength()

std::size_t TMVA::Experimental::SOFIE::ConvertShapeToLength ( std::vector< size_t > shape )

Definition at line 18 of file SOFIE_common.cxx.

◆ ConvertShapeToString()

std::string TMVA::Experimental::SOFIE::ConvertShapeToString ( std::vector< size_t > shape )

Definition at line 72 of file SOFIE_common.cxx.

◆ ConvertStringToType()

ETensorType TMVA::Experimental::SOFIE::ConvertStringToType ( std::string type )

Definition at line 57 of file SOFIE_common.cxx.

◆ ConvertTypeToString()

std::string TMVA::Experimental::SOFIE::ConvertTypeToString ( ETensorType type )

Definition at line 25 of file SOFIE_common.cxx.

◆ Copy()

GNN_Data TMVA::Experimental::SOFIE::Copy ( const GNN_Data & data )

inline

Definition at line 486 of file SOFIE_common.hxx.

◆ GetTemplatedType()

template<typename T >

ETensorType TMVA::Experimental::SOFIE::GetTemplatedType ( T )

Definition at line 112 of file SOFIE_common.hxx.

◆ operator|() [1/2]

std::underlying_type_t< Options > TMVA::Experimental::SOFIE::operator\|	(	Options	opA,
		Options	opB
	)

Definition at line 15 of file RModel.cxx.

◆ operator|() [2/2]

std::underlying_type_t< Options > TMVA::Experimental::SOFIE::operator\|	(	std::underlying_type_t< Options >	opA,
		Options	opB
	)

Definition at line 18 of file RModel.cxx.

◆ ParseBasicBinary()

template<EBasicBinaryOperator Op>

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::ParseBasicBinary	(	RModelParser_ONNX &	parser,
		const onnx::NodeProto &	nodeproto
	)

Definition at line 10 of file ParseBasicBinary.cxx.

◆ ParseBasicNary()

template<EBasicNaryOperator Op>

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::ParseBasicNary	(	RModelParser_ONNX &	parser,
		const onnx::NodeProto &	nodeproto
	)

Definition at line 11 of file ParseBasicNary.cxx.

◆ ParseBasicUnary()

template<EBasicUnaryOperator Op>

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::ParseBasicUnary	(	RModelParser_ONNX &	parser,
		const onnx::NodeProto &	nodeproto
	)

Definition at line 10 of file ParseBasicUnary.cxx.

◆ ParseReduce()

template<EReduceOpMode Op>

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::ParseReduce	(	RModelParser_ONNX &	parser,
		const onnx::NodeProto &	nodeproto
	)

Definition at line 11 of file ParseReduce.cxx.

Variable Documentation

◆ ParseAdd

ParserFuncSignature TMVA::Experimental::SOFIE::ParseAdd

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicBinary<EBasicBinaryOperator::Add>(parser, nodeproto);
}

Definition at line 52 of file ParseBasicBinary.cxx.

◆ ParseBatchNormalization

ParserFuncSignature TMVA::Experimental::SOFIE::ParseBatchNormalization

Definition at line 9 of file ParseBatchNormalization.cxx.

◆ ParseCast

ParserFuncSignature TMVA::Experimental::SOFIE::ParseCast

Definition at line 9 of file ParseCast.cxx.

◆ ParseConcat

ParserFuncSignature TMVA::Experimental::SOFIE::ParseConcat

Definition at line 9 of file ParseConcat.cxx.

◆ ParseConv

ParserFuncSignature TMVA::Experimental::SOFIE::ParseConv

Definition at line 9 of file ParseConv.cxx.

◆ ParseConvTranspose

ParserFuncSignature TMVA::Experimental::SOFIE::ParseConvTranspose

Definition at line 9 of file ParseConvTranspose.cxx.

◆ ParseDiv

ParserFuncSignature TMVA::Experimental::SOFIE::ParseDiv

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicBinary<EBasicBinaryOperator::Div>(parser, nodeproto);
}

Definition at line 67 of file ParseBasicBinary.cxx.

◆ ParseErf

ParserFuncSignature TMVA::Experimental::SOFIE::ParseErf

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser Erf op has input tensor" + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Erf<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Erf does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseErf.cxx.

◆ ParseExp

ParserFuncSignature TMVA::Experimental::SOFIE::ParseExp

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicUnary<EBasicUnaryOperator::kExp>(parser, nodeproto);
}

Definition at line 59 of file ParseBasicUnary.cxx.

◆ ParseExpand

ParserFuncSignature TMVA::Experimental::SOFIE::ParseExpand

Definition at line 9 of file ParseExpand.cxx.

◆ ParseFuseConvAdd

ParserFuseFuncSignature TMVA::Experimental::SOFIE::ParseFuseConvAdd

Initial value:

= [](RModelParser_ONNX & , const onnx::NodeProto & ,
                                              const onnx::NodeProto & ) {
   std::unique_ptr<ROperator> op;
   return op;
}

Definition at line 9 of file ParseFuseConvAdd.cxx.

◆ ParseFuseConvTransposeAdd

ParserFuseFuncSignature TMVA::Experimental::SOFIE::ParseFuseConvTransposeAdd

Definition at line 9 of file ParseFuseConvTransposeAdd.cxx.

◆ ParseFuseMatMulAdd

ParserFuseFuncSignature TMVA::Experimental::SOFIE::ParseFuseMatMulAdd

Definition at line 9 of file ParseFuseMatMulAdd.cxx.

◆ ParseGather

ParserFuncSignature TMVA::Experimental::SOFIE::ParseGather

Definition at line 10 of file ParseGather.cxx.

◆ ParseGemm

ParserFuncSignature TMVA::Experimental::SOFIE::ParseGemm

Definition at line 9 of file ParseGemm.cxx.

◆ ParseGRU

ParserFuncSignature TMVA::Experimental::SOFIE::ParseGRU

Definition at line 9 of file ParseGRU.cxx.

◆ ParseIdentity

ParserFuncSignature TMVA::Experimental::SOFIE::ParseIdentity

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type = ETensorType::UNDEFINED;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser Identity op has input tensor" + input_name +
                               "  but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Identity<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Identity does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseIdentity.cxx.

◆ ParseLayerNormalization

ParserFuncSignature TMVA::Experimental::SOFIE::ParseLayerNormalization

Definition at line 9 of file ParseLayerNormalization.cxx.

◆ ParseLeakyRelu

ParserFuncSignature TMVA::Experimental::SOFIE::ParseLeakyRelu

Definition at line 9 of file ParseLeakyRelu.cxx.

◆ ParseLog

ParserFuncSignature TMVA::Experimental::SOFIE::ParseLog

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicUnary<EBasicUnaryOperator::kLog>(parser, nodeproto);
}

Definition at line 64 of file ParseBasicUnary.cxx.

◆ ParseLSTM

ParserFuncSignature TMVA::Experimental::SOFIE::ParseLSTM

Definition at line 9 of file ParseLSTM.cxx.

◆ ParseMatMul

ParserFuncSignature TMVA::Experimental::SOFIE::ParseMatMul

Definition at line 9 of file ParseMatMul.cxx.

◆ ParseMax

ParserFuncSignature TMVA::Experimental::SOFIE::ParseMax

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicNary<EBasicNaryOperator::Max>(parser, nodeproto);
}

Definition at line 47 of file ParseBasicNary.cxx.

◆ ParseMean

ParserFuncSignature TMVA::Experimental::SOFIE::ParseMean

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicNary<EBasicNaryOperator::Mean>(parser, nodeproto);
}

Definition at line 55 of file ParseBasicNary.cxx.

◆ ParseMin

ParserFuncSignature TMVA::Experimental::SOFIE::ParseMin

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicNary<EBasicNaryOperator::Min>(parser, nodeproto);
}

Definition at line 51 of file ParseBasicNary.cxx.

◆ ParseMul

ParserFuncSignature TMVA::Experimental::SOFIE::ParseMul

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicBinary<EBasicBinaryOperator::Mul>(parser, nodeproto);
}

Definition at line 62 of file ParseBasicBinary.cxx.

◆ ParseNeg

ParserFuncSignature TMVA::Experimental::SOFIE::ParseNeg

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicUnary<EBasicUnaryOperator::kNeg>(parser, nodeproto);
}

Definition at line 54 of file ParseBasicUnary.cxx.

◆ ParsePool

ParserFuncSignature TMVA::Experimental::SOFIE::ParsePool

Definition at line 9 of file ParsePool.cxx.

◆ ParsePow

ParserFuncSignature TMVA::Experimental::SOFIE::ParsePow

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicBinary<EBasicBinaryOperator::Pow>(parser, nodeproto);
}

Definition at line 72 of file ParseBasicBinary.cxx.

◆ ParseReciprocal

ParserFuncSignature TMVA::Experimental::SOFIE::ParseReciprocal

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicUnary<EBasicUnaryOperator::kReciprocal>(parser, nodeproto);
}

Definition at line 49 of file ParseBasicUnary.cxx.

◆ ParseReduceMean

ParserFuncSignature TMVA::Experimental::SOFIE::ParseReduceMean

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseReduce<EReduceOpMode::ReduceMean>(parser, nodeproto);
}

Definition at line 63 of file ParseReduce.cxx.

◆ ParseReduceProd

ParserFuncSignature TMVA::Experimental::SOFIE::ParseReduceProd

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseReduce<EReduceOpMode::ReduceProd>(parser, nodeproto);
}

Definition at line 73 of file ParseReduce.cxx.

◆ ParseReduceSumsquare

ParserFuncSignature TMVA::Experimental::SOFIE::ParseReduceSumsquare

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseReduce<EReduceOpMode::ReduceSumsquare>(parser, nodeproto);
}

Definition at line 68 of file ParseReduce.cxx.

◆ ParseRelu

ParserFuncSignature TMVA::Experimental::SOFIE::ParseRelu

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser relu op has input tensor" + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Relu<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Relu does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseRelu.cxx.

◆ ParseReshape

ParserFuncSignature TMVA::Experimental::SOFIE::ParseReshape

Definition at line 9 of file ParseReshape.cxx.

◆ ParseRNN

ParserFuncSignature TMVA::Experimental::SOFIE::ParseRNN

Definition at line 9 of file ParseRNN.cxx.

◆ ParseSelu

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSelu

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type = ETensorType::UNDEFINED;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser selu op has input tensor" + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
 
   std::string output_name = nodeproto.output(0);
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Selu<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Selu does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseSelu.cxx.

◆ ParseShape

ParserFuncSignature TMVA::Experimental::SOFIE::ParseShape

Definition at line 9 of file ParseShape.cxx.

◆ ParseSigmoid

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSigmoid

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser Sigmoid op has input tensor" + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Sigmoid<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Sigmoid does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseSigmoid.cxx.

◆ ParseSlice

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSlice

Definition at line 9 of file ParseSlice.cxx.

◆ ParseSoftmax

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSoftmax

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser Softmax op has input tensor " + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   int64_t attr_axis = -1;
   if (nodeproto.attribute_size() == 1 && nodeproto.attribute(0).name() == "axis")
      attr_axis = nodeproto.attribute(0).i();
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Softmax<float>(attr_axis, input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Softmax does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
   return op;
}

Definition at line 9 of file ParseSoftmax.cxx.

◆ ParseSqrt

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSqrt

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicUnary<EBasicUnaryOperator::kSqrt>(parser, nodeproto);
}

Definition at line 44 of file ParseBasicUnary.cxx.

◆ ParseSub

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSub

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicBinary<EBasicBinaryOperator::Sub>(parser, nodeproto);
}

Definition at line 57 of file ParseBasicBinary.cxx.

◆ ParseSum

ParserFuncSignature TMVA::Experimental::SOFIE::ParseSum

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   return ParseBasicNary<EBasicNaryOperator::Sum>(parser, nodeproto);
}

Definition at line 59 of file ParseBasicNary.cxx.

◆ ParseTanh

ParserFuncSignature TMVA::Experimental::SOFIE::ParseTanh

Initial value:

= [](RModelParser_ONNX &parser, const onnx::NodeProto &nodeproto) {
   ETensorType input_type;
 
   auto input_name = nodeproto.input(0);
   if (parser.IsRegisteredTensorType(input_name)) {
      input_type = parser.GetTensorType(input_name);
   } else {
      throw std::runtime_error("TMVA::SOFIE ONNX Parser Tanh op has input tensor" + input_name +
                               " but its type is not yet registered");
   }
 
   std::unique_ptr<ROperator> op;
   std::string output_name = nodeproto.output(0);
 
   switch (input_type) {
   case ETensorType::FLOAT: op.reset(new ROperator_Tanh<float>(input_name, output_name)); break;
   default:
      throw std::runtime_error("TMVA::SOFIE - Unsupported - Operator Tanh does not yet support input type " +
                               std::to_string(static_cast<int>(input_type)));
   }
 
   if (!parser.IsRegisteredTensorType(output_name)) {
      parser.RegisterTensorType(output_name, input_type);
   }
 
   return op;
}

Definition at line 9 of file ParseTanh.cxx.

◆ ParseTranspose

ParserFuncSignature TMVA::Experimental::SOFIE::ParseTranspose

Definition at line 9 of file ParseTranspose.cxx.

Namespaces

Classes

Typedefs

Enumerations

Functions

Variables

Typedef Documentation

◆ int_t

◆ ParserFuncSignature

◆ ParserFuseFuncSignature

Enumeration Type Documentation

◆ Activation

◆ EBasicBinaryOperator

◆ EBasicNaryOperator

◆ EBasicUnaryOperator

◆ EReduceOpMode

◆ ETensorType

◆ FunctionReducer

◆ FunctionRelation

◆ FunctionTarget

◆ FunctionType

◆ GraphType

◆ Options

◆ PoolOpMode

◆ ReshapeOpMode

◆ WeightFileType

Function Documentation

◆ Concatenate() [1/2]

◆ Concatenate() [2/2]

◆ ConvertShapeToDim()

◆ ConvertShapeToLength()

◆ ConvertShapeToString()

◆ ConvertStringToType()

◆ ConvertTypeToString()

◆ Copy()

◆ GetTemplatedType()

◆ operator|() [1/2]

◆ operator|() [2/2]

◆ ParseBasicBinary()

◆ ParseBasicNary()

◆ ParseBasicUnary()

◆ ParseReduce()

Variable Documentation

◆ ParseAdd

◆ ParseBatchNormalization

◆ ParseCast

◆ ParseConcat

◆ ParseConv

◆ ParseConvTranspose

◆ ParseDiv

◆ ParseErf

◆ ParseExp

◆ ParseExpand

◆ ParseFuseConvAdd

◆ ParseFuseConvTransposeAdd

◆ ParseFuseMatMulAdd

◆ ParseGather

◆ ParseGemm

◆ ParseGRU

◆ ParseIdentity

◆ ParseLayerNormalization

◆ ParseLeakyRelu

◆ ParseLog

◆ ParseLSTM

◆ ParseMatMul

◆ ParseMax

◆ ParseMean

◆ ParseMin

◆ ParseMul

◆ ParseNeg

◆ ParsePool

◆ ParsePow

◆ ParseReciprocal

◆ ParseReduceMean

◆ ParseReduceProd

◆ ParseReduceSumsquare

◆ ParseRelu

◆ ParseReshape

◆ ParseRNN

◆ ParseSelu