template<typename T>
class TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >

Definition at line 15 of file ROperator_LayerNormalization.hxx.

Public Member Functions
	ROperator_LayerNormalization ()

	ROperator_LayerNormalization (int axis, float epsilon, size_t stashType, const std::string &nameX, const std::string &nameScale, const std::string &nameB, const std::string &nameY, const std::string &nameMean, const std::string &nameInvStdDev)

std::string	Generate (std::string opName) override

std::string	GenerateInitCode () override

std::vector< std::string >	GetBlasRoutines () override

std::vector< std::string >	GetStdLibs () override

void	Initialize (RModel &model) override

std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > > input) override

std::vector< ETensorType >	TypeInference (std::vector< ETensorType > input) override

Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual	~ROperator ()

virtual std::string	GenerateDeclCode ()

virtual std::string	GenerateSessionMembersCode (std::string)

std::span< const std::string_view >	GetOpInputTensors () const

std::span< const std::string_view >	GetOpOutputTensors () const

virtual std::string	Header ()

Private Attributes
int	fAttrAxis

float	fAttrEpsilon

size_t	fAttrStashType

std::string	fAxesLength

std::vector< Dim >	fAxesShape

size_t	fAxis

std::string	fLength

std::string	fNB

std::string	fNBroadcastedB

std::string	fNCastedX

std::string	fNInvStdDev

std::string	fNMean

std::string	fNNormalizedX

std::string	fNormalizedLength

std::vector< Dim >	fNormalizedShape

std::string	fNScale

std::string	fNX

std::string	fNY

std::vector< size_t >	fShapeB

std::vector< Dim >	fShapeInvStdDev

std::vector< Dim >	fShapeMean

std::vector< Dim >	fShapeScale

std::vector< Dim >	fShapeX

std::vector< Dim >	fShapeY

size_t	fSize

std::string	fType

Additional Inherited Members
Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator
std::vector< std::string_view >	fInputTensorNames

bool	fIsOutputConstant = false
	flag to identify if operator has a constant output (no need to generate code)

bool	fIsOutputParamShape = false
	flag to identify of the output represents a parametric shape (can be knwon at compile time)

std::vector< std::string_view >	fOutputTensorNames

bool	fUseSession = false
	flag to identify if using the session class

const std::string	SP = " "
	space used to correctly indent the generated C++ code

#include <TMVA/ROperator_LayerNormalization.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >:

[legend]

Constructor & Destructor Documentation

◆ ROperator_LayerNormalization() [1/2]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::ROperator_LayerNormalization ( )

inline

Definition at line 53 of file ROperator_LayerNormalization.hxx.

◆ ROperator_LayerNormalization() [2/2]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::ROperator_LayerNormalization	(	int	axis,
		float	epsilon,
		size_t	stashType,
		const std::string &	nameX,
		const std::string &	nameScale,
		const std::string &	nameB,
		const std::string &	nameY,
		const std::string &	nameMean,
		const std::string &	nameInvStdDev )

inline

Definition at line 55 of file ROperator_LayerNormalization.hxx.

Member Function Documentation

◆ Generate()

template<typename T >

std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::Generate ( std::string opName )

inlineoverridevirtual

Implements TMVA::Experimental::SOFIE::ROperator.

Definition at line 158 of file ROperator_LayerNormalization.hxx.

◆ GenerateInitCode()

template<typename T >

std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GenerateInitCode ( )

inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 143 of file ROperator_LayerNormalization.hxx.

◆ GetBlasRoutines()

template<typename T >

std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GetBlasRoutines ( )

inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 336 of file ROperator_LayerNormalization.hxx.

◆ GetStdLibs()

template<typename T >

std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GetStdLibs ( )

inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 338 of file ROperator_LayerNormalization.hxx.

◆ Initialize()

template<typename T >

void TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::Initialize ( RModel & model )

inlineoverridevirtual

Implements TMVA::Experimental::SOFIE::ROperator.

Definition at line 80 of file ROperator_LayerNormalization.hxx.

◆ ShapeInference()

template<typename T >

std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::ShapeInference ( std::vector< std::vector< size_t > > input )