Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
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< RModelGetFunctionBlock ()
 
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< RModelfunction_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:
[legend]

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.

  • tmva/sofie/inc/TMVA/RFunction_MLP.hxx
  • tmva/sofie/src/RFunction_MLP.cxx