TMVA::Experimental::SOFIE::RModel_GNN Class Reference

Definition at line 87 of file RModel_GNN.hxx.

Public Member Functions
	RModel_GNN (GNN_Init &graph_input_struct)
void	Generate () final
Public Member Functions inherited from TMVA::Experimental::SOFIE::RModel_GNNBase
virtual	~RModel_GNNBase ()=default
Public Member Functions inherited from TMVA::Experimental::SOFIE::RModel_Base
	RModel_Base ()=default
	Default constructor.
	RModel_Base (std::string function_name)
	RModel_Base (std::string name, std::string parsedtime)
void	AddBlasRoutines (std::vector< std::string > routines)
void	AddNeededCustomHeader (std::string filename)
void	AddNeededStdLib (std::string libname)
void	GenerateHeaderInfo (std::string &hgname)
std::string	GetFilename ()
const std::string &	GetName () const
void	OutputGenerated (std::string filename="", bool append=false)
void	PrintGenerated ()
std::string	ReturnGenerated ()
void	SetFilename (std::string filename)

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

Additional Inherited Members
Protected Attributes inherited from TMVA::Experimental::SOFIE::RModel_Base
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

#include <TMVA/RModel_GNN.hxx>

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

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

Generate()

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

finalvirtual

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

Definition at line 36 of file RModel_GNN.cxx.

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.

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.