TMVA::Experimental::SOFIE::RFunction_MLP Class Reference

Definition at line 18 of file RFunction_MLP.hxx.

Public Member Functions
	RFunction_MLP (FunctionTarget target, Int_t numLayers, Activation activation_function=Activation::RELU, bool activate_final=false, GraphType gType=GraphType::GNN)
virtual	~RFunction_MLP ()
void	AddInitializedTensors (const std::vector< std::vector< std::string > > &initialized_tensors)
void	AddLayerNormalization (int axis, float epsilon, size_t stashType, const std::string &nameX, const std::string &nameScale, const std::string &nameB, const std::string &nameY)
void	Initialize ()
Public Member Functions inherited from TMVA::Experimental::SOFIE::RFunction_Update
	RFunction_Update ()
	RFunction_Update (FunctionTarget target, GraphType gType)
virtual	~RFunction_Update ()
void	AddInputTensors (const std::vector< std::vector< Dim > > &inputShapes)
void	AddInputTensors (const std::vector< std::vector< std::size_t > > &inputShapes)
std::string	Generate (const std::vector< std::string > &inputPtrs)
std::string	GenerateModel (const std::string &filename, long read_pos=0, long block_size=-1)
std::shared_ptr< RModel >	GetFunctionBlock ()
FunctionTarget	GetFunctionTarget ()
Public Member Functions inherited from TMVA::Experimental::SOFIE::RFunction
	RFunction ()
	RFunction (std::string funcName, FunctionType type)
virtual	~RFunction ()
FunctionType	GetFunctionType ()

Private Attributes
bool	fActivateFinal
Activation	fActivationFunction
std::vector< std::string >	fBiasTensors
std::vector< std::string >	fKernelTensors
Int_t	fNumLayers

Additional Inherited Members
Protected Attributes inherited from TMVA::Experimental::SOFIE::RFunction_Update
std::vector< ROperator * >	fAddlOp
GraphType	fGraphType
std::vector< std::string >	fInputTensors
FunctionTarget	fTarget
std::shared_ptr< RModel >	function_block
Protected Attributes inherited from TMVA::Experimental::SOFIE::RFunction
std::string	fFuncName
FunctionType	fType

#include <TMVA/RFunction_MLP.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::RFunction_MLP:

Constructor & Destructor Documentation

~RFunction_MLP()

virtual TMVA::Experimental::SOFIE::RFunction_MLP::~RFunction_MLP ( )

inlinevirtual

Definition at line 27 of file RFunction_MLP.hxx.

RFunction_MLP()

TMVA::Experimental::SOFIE::RFunction_MLP::RFunction_MLP	(	FunctionTarget	target,
		Int_t	numLayers,
		Activation	activation_function = Activation::RELU,
		bool	activate_final = false,
		GraphType	gType = GraphType::GNN
	)

Definition at line 14 of file RFunction_MLP.cxx.

Member Function Documentation

AddInitializedTensors()

void TMVA::Experimental::SOFIE::RFunction_MLP::AddInitializedTensors ( const std::vector< std::vector< std::string > > &  initialized_tensors )

inlinevirtual

Reimplemented from TMVA::Experimental::SOFIE::RFunction_Update.

Definition at line 35 of file RFunction_MLP.hxx.

AddLayerNormalization()

void TMVA::Experimental::SOFIE::RFunction_MLP::AddLayerNormalization ( int  axis, float  epsilon, size_t  stashType, const std::string &  nameX, const std::string &  nameScale, const std::string &  nameB, const std::string &  nameY  )

virtual

Reimplemented from TMVA::Experimental::SOFIE::RFunction_Update.

Definition at line 76 of file RFunction_MLP.cxx.

Initialize()

void TMVA::Experimental::SOFIE::RFunction_MLP::Initialize ( )

virtual

Reimplemented from TMVA::Experimental::SOFIE::RFunction_Update.

Definition at line 29 of file RFunction_MLP.cxx.

Member Data Documentation

fActivateFinal

bool TMVA::Experimental::SOFIE::RFunction_MLP::fActivateFinal

private

Definition at line 22 of file RFunction_MLP.hxx.

fActivationFunction

Activation TMVA::Experimental::SOFIE::RFunction_MLP::fActivationFunction

private

Definition at line 21 of file RFunction_MLP.hxx.

fBiasTensors

std::vector<std::string> TMVA::Experimental::SOFIE::RFunction_MLP::fBiasTensors

private

Definition at line 24 of file RFunction_MLP.hxx.

fKernelTensors

std::vector<std::string> TMVA::Experimental::SOFIE::RFunction_MLP::fKernelTensors

private

Definition at line 23 of file RFunction_MLP.hxx.

fNumLayers

Int_t TMVA::Experimental::SOFIE::RFunction_MLP::fNumLayers

private

Definition at line 20 of file RFunction_MLP.hxx.