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

Definition at line 17 of file ROperator_BatchNormalization.hxx.

Public Member Functions
	ROperator_BatchNormalization ()=delete

	ROperator_BatchNormalization (float epsilon, float momentum, std::size_t training_mode, std::string nameX, std::string nameScale, std::string nameB, std::string nameMean, std::string nameVar, std::string nameY, EActivationType activation=EActivationType::UNDEFINED)

std::string	Generate (std::string OpName)

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

void	Initialize (RModel &model) override

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

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

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

virtual std::string	GenerateDeclCode ()

virtual std::string	GenerateInitCode ()

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

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

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

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

virtual std::string	Header ()

Private Attributes
EActivationType	fActivation

float	fepsilon = 1e-05

float	fmomentum = 0.9

std::string	fNB

std::string	fNMean

std::string	fNScale

std::string	fNVar

std::string	fNX

std::string	fNY

std::vector< size_t >	fShapeB

std::vector< size_t >	fShapeMean

std::vector< size_t >	fShapeScale

std::vector< size_t >	fShapeVar

std::vector< size_t >	fShapeX

std::vector< size_t >	fShapeY

std::size_t	ftraining_mode = 0

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)

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_BatchNormalization.hxx>

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

[legend]

Constructor & Destructor Documentation

◆ ROperator_BatchNormalization() [1/2]

template<typename T >

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

delete

◆ ROperator_BatchNormalization() [2/2]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_BatchNormalization< T >::ROperator_BatchNormalization	(	float	epsilon,
		float	momentum,
		std::size_t	training_mode,
		std::string	nameX,
		std::string	nameScale,
		std::string	nameB,
		std::string	nameMean,
		std::string	nameVar,
		std::string	nameY,
		EActivationType	activation = EActivationType::UNDEFINED )

inline

Definition at line 48 of file ROperator_BatchNormalization.hxx.

Member Function Documentation

◆ Generate()

template<typename T >

std::string TMVA::Experimental::SOFIE::ROperator_BatchNormalization< T >::Generate ( std::string OpName )

inlinevirtual

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

Definition at line 187 of file ROperator_BatchNormalization.hxx.

◆ GetBlasRoutines()

template<typename T >

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

inlinevirtual

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

Definition at line 232 of file ROperator_BatchNormalization.hxx.

◆ Initialize()

template<typename T >

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

inlineoverridevirtual

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

Definition at line 90 of file ROperator_BatchNormalization.hxx.

◆ ShapeInference()

template<typename T >

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

inlinevirtual

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

Definition at line 74 of file ROperator_BatchNormalization.hxx.