Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
List of all members | Public Member Functions | Private Attributes | List of all members
TMVA::Experimental::SOFIE::ROperator_If Class Referencefinal

Definition at line 16 of file ROperator_SubGraph.hxx.

Public Member Functions

 ROperator_If ()
 
 ROperator_If (const std::string &nameX, const std::vector< std::string > &nameYs, std::unique_ptr< RModel > model_then, std::unique_ptr< RModel > model_else)
 
std::string Generate (std::string opName) override
 
void Initialize (RModel &model) override
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input) override
 
std::vector< ETensorType > TypeInference (std::vector< ETensorType > input) override
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual ~ROperator ()
 
virtual std::string GenerateDeclCode ()
 
virtual std::string GenerateInitCode ()
 
virtual std::string GenerateSessionMembersCode (std::string)
 
virtual std::vector< std::string > GetBlasRoutines ()
 
std::span< const std::string_view > GetOpInputTensors () const
 
std::span< const std::string_view > GetOpOutputTensors () const
 
virtual std::vector< std::string > GetStdLibs ()
 
virtual std::string Header ()
 

Private Attributes

std::string fInputSignature_modelElse
 
std::string fInputSignature_modelThen
 
std::shared_ptr< RModel > fModel_else
 
std::shared_ptr< RModel > fModel_then
 
std::string fNX
 
std::vector< std::string > fNYs
 
ETensorType fType = ETensorType::UNDEFINED
 

Additional Inherited Members

- Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator
std::vector< std::string_view > fInputTensorNames
 
bool fIsOutputConstant = false
 flag to identify if operator has a constant output (no need to generate code)
 
std::vector< std::string_view > fOutputTensorNames
 
bool fUseSession = false
 flag to identify if using the session class
 
const std::string SP = " "
 space used to correctly indent the generated C++ code
 

#include <TMVA/ROperator_SubGraph.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_If:
[legend]

Constructor & Destructor Documentation

◆ ROperator_If() [1/2]

TMVA::Experimental::SOFIE::ROperator_If::ROperator_If ( )
inline

Definition at line 30 of file ROperator_SubGraph.hxx.

◆ ROperator_If() [2/2]

TMVA::Experimental::SOFIE::ROperator_If::ROperator_If ( const std::string & nameX,
const std::vector< std::string > & nameYs,
std::unique_ptr< RModel > model_then,
std::unique_ptr< RModel > model_else )
inline

Definition at line 31 of file ROperator_SubGraph.hxx.

Member Function Documentation

◆ Generate()

std::string TMVA::Experimental::SOFIE::ROperator_If::Generate ( std::string opName)
inlineoverridevirtual

Implements TMVA::Experimental::SOFIE::ROperator.

Definition at line 83 of file ROperator_SubGraph.hxx.

◆ Initialize()

void TMVA::Experimental::SOFIE::ROperator_If::Initialize ( RModel & model)
inlineoverridevirtual

Implements TMVA::Experimental::SOFIE::ROperator.

Definition at line 51 of file ROperator_SubGraph.hxx.

◆ ShapeInference()

std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_If::ShapeInference ( std::vector< std::vector< size_t > > input)
inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 46 of file ROperator_SubGraph.hxx.

◆ TypeInference()

std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_If::TypeInference ( std::vector< ETensorType > input)
inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 42 of file ROperator_SubGraph.hxx.

Member Data Documentation

◆ fInputSignature_modelElse

std::string TMVA::Experimental::SOFIE::ROperator_If::fInputSignature_modelElse
private

Definition at line 27 of file ROperator_SubGraph.hxx.

◆ fInputSignature_modelThen

std::string TMVA::Experimental::SOFIE::ROperator_If::fInputSignature_modelThen
private

Definition at line 26 of file ROperator_SubGraph.hxx.

◆ fModel_else

std::shared_ptr<RModel> TMVA::Experimental::SOFIE::ROperator_If::fModel_else
private

Definition at line 25 of file ROperator_SubGraph.hxx.

◆ fModel_then

std::shared_ptr<RModel> TMVA::Experimental::SOFIE::ROperator_If::fModel_then
private

Definition at line 24 of file ROperator_SubGraph.hxx.

◆ fNX

std::string TMVA::Experimental::SOFIE::ROperator_If::fNX
private

Definition at line 21 of file ROperator_SubGraph.hxx.

◆ fNYs

std::vector<std::string> TMVA::Experimental::SOFIE::ROperator_If::fNYs
private

Definition at line 23 of file ROperator_SubGraph.hxx.

◆ fType

ETensorType TMVA::Experimental::SOFIE::ROperator_If::fType = ETensorType::UNDEFINED
private

Definition at line 22 of file ROperator_SubGraph.hxx.

  • tmva/sofie/inc/TMVA/ROperator_SubGraph.hxx