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) override
virtual std::string	GenerateDeclCode ()
virtual std::string	GenerateInitCode ()
virtual std::string	GenerateSessionMembersCode (std::string)
std::vector< std::string >	GetBlasRoutines () override
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 ()
void	Initialize (RModel &model) override
bool	IsOutputConstant () const
	check if the output of the operator is Constant and is evaluated at initialization time
std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > > input) override
std::vector< ETensorType >	TypeInference (std::vector< ETensorType > input) override

Protected Attributes
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

Private Attributes
EActivationType	fActivation
float	fepsilon = 1e-05
float	fmomentum = 0.9
std::string	fNB
std::string	fNFusedScale
std::string	fNMean
std::string	fNScale
std::string	fNVar
std::string	fNX
std::string	fNY
std::vector< Dim >	fShapeX
std::vector< Dim >	fShapeY
std::size_t	ftraining_mode = 0
std::string	fType

#include <TMVA/ROperator_BatchNormalization.hxx>

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

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 45 of file ROperator_BatchNormalization.hxx.

Member Function Documentation

◆ Generate()

template<typename T>

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

inlineoverridevirtual

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

Definition at line 144 of file ROperator_BatchNormalization.hxx.

◆ GenerateDeclCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateDeclCode ( )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_Random.

Definition at line 31 of file ROperator.hxx.

◆ GenerateInitCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateInitCode ( )

inlinevirtualinherited

Definition at line 29 of file ROperator.hxx.

◆ GenerateSessionMembersCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateSessionMembersCode ( std::string )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_LSTM< T >, TMVA::Experimental::SOFIE::ROperator_NonZero< T >, TMVA::Experimental::SOFIE::ROperator_Pool< T >, and TMVA::Experimental::SOFIE::ROperator_RNN< T >.

Definition at line 33 of file ROperator.hxx.

◆ GetBlasRoutines()

template<typename T>

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

inlineoverridevirtual

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

Definition at line 186 of file ROperator_BatchNormalization.hxx.

◆ GetOpInputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpInputTensors ( ) const

inlineinherited

Definition at line 54 of file ROperator.hxx.

◆ GetOpOutputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpOutputTensors ( ) const

inlineinherited

Definition at line 58 of file ROperator.hxx.

◆ GetStdLibs()

virtual std::vector< std::string > TMVA::Experimental::SOFIE::ROperator::GetStdLibs ( )

inlinevirtualinherited

Definition at line 23 of file ROperator.hxx.

◆ Header()

virtual std::string TMVA::Experimental::SOFIE::ROperator::Header ( )

inlinevirtualinherited

Definition at line 34 of file ROperator.hxx.

◆ Initialize()

template<typename T>

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

inlineoverridevirtual

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

Definition at line 88 of file ROperator_BatchNormalization.hxx.

◆ IsOutputConstant()

bool TMVA::Experimental::SOFIE::ROperator::IsOutputConstant ( ) const

inlineinherited

check if the output of the operator is Constant and is evaluated at initialization time

Definition at line 37 of file ROperator.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 )