Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
List of all members | Public Member Functions | Protected Attributes | Private Attributes | List of all members
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) override
 
void AddInputTensors (const std::vector< std::vector< Dim > > &inputShapes)
 
void AddInputTensors (const std::vector< std::vector< std::size_t > > &inputShapes)
 
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) override
 
std::string Generate (const std::vector< std::string > &inputPtrs)
 
std::string GenerateModel (const std::string &filename, long read_pos=0, long block_size=-1, bool verbose=false)
 
std::shared_ptr< RModel > GetFunctionBlock ()
 
FunctionTarget GetFunctionTarget ()
 
FunctionType GetFunctionType ()
 
void Initialize () override
 

Protected Attributes

std::vector< ROperator * > fAddlOp
 
std::string fFuncName
 
std::shared_ptr< RModel > fFunction_block
 
GraphType fGraphType
 
std::vector< std::string > fInputTensors
 
FunctionTarget fTarget
 
FunctionType fType
 

Private Attributes

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

#include <TMVA/RFunction_MLP.hxx>

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

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)
inlineoverridevirtual

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

Definition at line 35 of file RFunction_MLP.hxx.

◆ AddInputTensors() [1/2]

void TMVA::Experimental::SOFIE::RFunction_Update::AddInputTensors ( const std::vector< std::vector< Dim > > & inputShapes)
inherited

Definition at line 56 of file RFunction.cxx.

◆ AddInputTensors() [2/2]

void TMVA::Experimental::SOFIE::RFunction_Update::AddInputTensors ( const std::vector< std::vector< std::size_t > > & inputShapes)
inherited

Definition at line 50 of file RFunction.cxx.

◆ 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 )
overridevirtual

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

Definition at line 76 of file RFunction_MLP.cxx.

◆ Generate()

std::string TMVA::Experimental::SOFIE::RFunction_Update::Generate ( const std::vector< std::string > & inputPtrs)
inherited

Definition at line 72 of file RFunction.cxx.

◆ GenerateModel()

std::string TMVA::Experimental::SOFIE::RFunction_Update::GenerateModel ( const std::string & filename,
long read_pos = 0,
long block_size = -1,
bool verbose = false )
inherited

Definition at line 63 of file RFunction.cxx.

◆ GetFunctionBlock()

std::shared_ptr< RModel > TMVA::Experimental::SOFIE::RFunction_Update::GetFunctionBlock ( )
inlineinherited

Definition at line 52 of file RFunction.hxx.

◆ GetFunctionTarget()

FunctionTarget TMVA::Experimental::SOFIE::RFunction_Update::GetFunctionTarget ( )
inlineinherited

Definition at line 57 of file RFunction.hxx.

◆ GetFunctionType()

FunctionType TMVA::Experimental::SOFIE::RFunction::GetFunctionType ( )
inlineinherited

Definition at line 24 of file RFunction.hxx.

◆ Initialize()

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

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.

◆ fAddlOp

std::vector<ROperator*> TMVA::Experimental::SOFIE::RFunction_Update::fAddlOp
protectedinherited

Definition at line 39 of file RFunction.hxx.

◆ fBiasTensors

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

Definition at line 24 of file RFunction_MLP.hxx.

◆ fFuncName

std::string TMVA::Experimental::SOFIE::RFunction::fFuncName
protectedinherited

Definition at line 19 of file RFunction.hxx.

◆ fFunction_block

std::shared_ptr<RModel> TMVA::Experimental::SOFIE::RFunction_Update::fFunction_block
protectedinherited

Definition at line 35 of file RFunction.hxx.

◆ fGraphType

GraphType TMVA::Experimental::SOFIE::RFunction_Update::fGraphType
protectedinherited

Definition at line 37 of file RFunction.hxx.

◆ fInputTensors

std::vector<std::string> TMVA::Experimental::SOFIE::RFunction_Update::fInputTensors
protectedinherited

Definition at line 38 of file RFunction.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.

◆ fTarget

FunctionTarget TMVA::Experimental::SOFIE::RFunction_Update::fTarget
protectedinherited

Definition at line 36 of file RFunction.hxx.

◆ fType

FunctionType TMVA::Experimental::SOFIE::RFunction::fType
protectedinherited

Definition at line 20 of file RFunction.hxx.

The documentation for this class was generated from the following files: