TMVA::Experimental::SOFIE::ROperator_If Class Reference

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:

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

Implements 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

Implements 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.