Logo ROOT  

Reference Guide
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::RModel_GNN Class Referencefinal

Definition at line 87 of file RModel_GNN.hxx.

Public Member Functions

 RModel_GNN (GNN_Init &graph_input_struct)
void AddBlasRoutines (std::vector< std::string > routines)
void AddNeededCustomHeader (std::string filename)
void AddNeededStdLib (std::string libname)
void Generate () final
void GenerateHeaderInfo (std::string &hgname)
std::string GetFilename ()
const std::string & GetName () const
void OutputGenerated (std::string filename="", bool append=false)
void PrintGenerated (std::ostream &os=std::cout)
std::string ReturnGenerated ()
void SetFilename (std::string filename)

Protected Attributes

std::unordered_set< std::string > fCustomOpHeaders
std::string fFileName
std::string fGC
bool fIsGNN = false
bool fIsGNNComponent = false
std::string fName = "UnnamedModel"
std::unordered_set< std::string > fNeededBlasRoutines
std::unordered_set< std::string > fNeededStdLib = {"vector"}
std::string fParseTime
bool fUseSession = true
bool fUseWeightFile = true
WeightFileType fWeightFile = WeightFileType::Text

Private Attributes

std::unique_ptr< RFunction_Aggregate > edge_global_agg_block
std::unique_ptr< RFunction_Aggregate > edge_node_agg_block
std::unique_ptr< RFunction_Update > edges_update_block
std::unique_ptr< RFunction_Update > globals_update_block
std::unique_ptr< RFunction_Aggregate > node_global_agg_block
std::unique_ptr< RFunction_Update > nodes_update_block
std::size_t num_edge_features
std::size_t num_edges
std::size_t num_global_features
std::size_t num_node_features
std::size_t num_nodes

#include <TMVA/RModel_GNN.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::RModel_GNN:
TMVA::Experimental::SOFIE::RModel_GNNBase TMVA::Experimental::SOFIE::RModel_Base

Constructor & Destructor Documentation

◆ RModel_GNN()

TMVA::Experimental::SOFIE::RModel_GNN::RModel_GNN ( GNN_Init & graph_input_struct)

Definition at line 13 of file RModel_GNN.cxx.

Member Function Documentation

◆ AddBlasRoutines()

void TMVA::Experimental::SOFIE::RModel_Base::AddBlasRoutines ( std::vector< std::string > routines)
inlineinherited

Definition at line 77 of file RModel_Base.hxx.

◆ AddNeededCustomHeader()

void TMVA::Experimental::SOFIE::RModel_Base::AddNeededCustomHeader ( std::string filename)
inlineinherited

Definition at line 90 of file RModel_Base.hxx.

◆ AddNeededStdLib()

void TMVA::Experimental::SOFIE::RModel_Base::AddNeededStdLib ( std::string libname)
inlineinherited

Definition at line 83 of file RModel_Base.hxx.

◆ Generate()

void TMVA::Experimental::SOFIE::RModel_GNN::Generate ( )
finalvirtual

Implements TMVA::Experimental::SOFIE::RModel_GNNBase.

Definition at line 36 of file RModel_GNN.cxx.

◆ GenerateHeaderInfo()

void TMVA::Experimental::SOFIE::RModel_Base::GenerateHeaderInfo ( std::string & hgname)
inherited

Definition at line 15 of file RModel_Base.cxx.

◆ GetFilename()

std::string TMVA::Experimental::SOFIE::RModel_Base::GetFilename ( )
inlineinherited

Definition at line 100 of file RModel_Base.hxx.

◆ GetName()

const std::string & TMVA::Experimental::SOFIE::RModel_Base::GetName ( ) const
inlineinherited

Definition at line 101 of file RModel_Base.hxx.

◆ OutputGenerated()

void TMVA::Experimental::SOFIE::RModel_Base::OutputGenerated ( std::string filename = "",
bool append = false )
inherited

Definition at line 62 of file RModel_Base.cxx.

◆ PrintGenerated()

void TMVA::Experimental::SOFIE::RModel_Base::PrintGenerated ( std::ostream & os = std::cout)
inlineinherited

Definition at line 95 of file RModel_Base.hxx.

◆ ReturnGenerated()

std::string TMVA::Experimental::SOFIE::RModel_Base::ReturnGenerated ( )
inlineinherited

Definition at line 97 of file RModel_Base.hxx.

◆ SetFilename()

void TMVA::Experimental::SOFIE::RModel_Base::SetFilename ( std::string filename)
inlineinherited

Definition at line 99 of file RModel_Base.hxx.

Member Data Documentation

◆ edge_global_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::edge_global_agg_block
private

Definition at line 97 of file RModel_GNN.hxx.

◆ edge_node_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::edge_node_agg_block
private

Definition at line 96 of file RModel_GNN.hxx.

◆ edges_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::edges_update_block
private

Definition at line 91 of file RModel_GNN.hxx.

◆ fCustomOpHeaders

std::unordered_set<std::string> TMVA::Experimental::SOFIE::RModel_Base::fCustomOpHeaders
protectedinherited

Definition at line 56 of file RModel_Base.hxx.

◆ fFileName

std::string TMVA::Experimental::SOFIE::RModel_Base::fFileName
protectedinherited

Definition at line 48 of file RModel_Base.hxx.

◆ fGC

std::string TMVA::Experimental::SOFIE::RModel_Base::fGC
protectedinherited

Definition at line 59 of file RModel_Base.hxx.

◆ fIsGNN

bool TMVA::Experimental::SOFIE::RModel_Base::fIsGNN = false
protectedinherited

Definition at line 62 of file RModel_Base.hxx.

◆ fIsGNNComponent

bool TMVA::Experimental::SOFIE::RModel_Base::fIsGNNComponent = false
protectedinherited

Definition at line 63 of file RModel_Base.hxx.

◆ fName

std::string TMVA::Experimental::SOFIE::RModel_Base::fName = "UnnamedModel"
protectedinherited

Definition at line 58 of file RModel_Base.hxx.

◆ fNeededBlasRoutines

std::unordered_set<std::string> TMVA::Experimental::SOFIE::RModel_Base::fNeededBlasRoutines
protectedinherited

Definition at line 53 of file RModel_Base.hxx.

◆ fNeededStdLib

std::unordered_set<std::string> TMVA::Experimental::SOFIE::RModel_Base::fNeededStdLib = {"vector"}
protectedinherited

Definition at line 55 of file RModel_Base.hxx.

◆ fParseTime

std::string TMVA::Experimental::SOFIE::RModel_Base::fParseTime
protectedinherited

Definition at line 49 of file RModel_Base.hxx.

◆ fUseSession

bool TMVA::Experimental::SOFIE::RModel_Base::fUseSession = true
protectedinherited

Definition at line 61 of file RModel_Base.hxx.

◆ fUseWeightFile

bool TMVA::Experimental::SOFIE::RModel_Base::fUseWeightFile = true
protectedinherited

Definition at line 60 of file RModel_Base.hxx.

◆ fWeightFile

WeightFileType TMVA::Experimental::SOFIE::RModel_Base::fWeightFile = WeightFileType::Text
protectedinherited

Definition at line 51 of file RModel_Base.hxx.

◆ globals_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::globals_update_block
private

Definition at line 93 of file RModel_GNN.hxx.

◆ node_global_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::node_global_agg_block
private

Definition at line 98 of file RModel_GNN.hxx.

◆ nodes_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::nodes_update_block
private

Definition at line 92 of file RModel_GNN.hxx.

◆ num_edge_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_edge_features
private

Definition at line 104 of file RModel_GNN.hxx.

◆ num_edges

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_edges
private

Definition at line 101 of file RModel_GNN.hxx.

◆ num_global_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_global_features
private

Definition at line 105 of file RModel_GNN.hxx.

◆ num_node_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_node_features
private

Definition at line 103 of file RModel_GNN.hxx.

◆ num_nodes

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_nodes
private

Definition at line 100 of file RModel_GNN.hxx.

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