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_Custom< T > Class Template Referencefinal

template<typename T>
class TMVA::Experimental::SOFIE::ROperator_Custom< T >

Definition at line 15 of file ROperator_Custom.hxx.

Public Member Functions

 ROperator_Custom ()
 
 ROperator_Custom (std::string OpName, std::vector< std::string >Inputs, std::vector< std::string >Outputs, std::vector< std::vector< std::size_t > > OutputShapes, std::string HeaderName)
 
std::string Generate (std::string OpName)
 
void Initialize (RModel &model) override
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > >)
 
std::vector< ETensorType > TypeInference (std::vector< ETensorType >)
 
- 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 fHeaderName
 
std::vector< std::string > fInputNames
 
std::vector< std::size_t > fInputSizes
 
ETensorType fInputType
 
std::string fOpName
 
std::vector< std::string > fOutputNames
 
std::vector< std::vector< std::size_t > > fOutputShapes
 

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_Custom.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Custom< T >:
[legend]

Constructor & Destructor Documentation

◆ ROperator_Custom() [1/2]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Custom< T >::ROperator_Custom ( )
inline

Definition at line 28 of file ROperator_Custom.hxx.

◆ ROperator_Custom() [2/2]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Custom< T >::ROperator_Custom ( std::string OpName,
std::vector< std::string > Inputs,
std::vector< std::string > Outputs,
std::vector< std::vector< std::size_t > > OutputShapes,
std::string HeaderName )
inline

Definition at line 29 of file ROperator_Custom.hxx.

Member Function Documentation

◆ Generate()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Custom< T >::Generate ( std::string OpName)
inlinevirtual

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

Definition at line 78 of file ROperator_Custom.hxx.

◆ Initialize()

template<typename T >
void TMVA::Experimental::SOFIE::ROperator_Custom< T >::Initialize ( RModel & model)
inlineoverridevirtual

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

Definition at line 46 of file ROperator_Custom.hxx.

◆ ShapeInference()

template<typename T >
std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_Custom< T >::ShapeInference ( std::vector< std::vector< size_t > > )
inlinevirtual

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

Definition at line 43 of file ROperator_Custom.hxx.

◆ TypeInference()

template<typename T >
std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_Custom< T >::TypeInference ( std::vector< ETensorType > )
inlinevirtual

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

Definition at line 44 of file ROperator_Custom.hxx.

Member Data Documentation

◆ fHeaderName

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Custom< T >::fHeaderName
private

Definition at line 24 of file ROperator_Custom.hxx.

◆ fInputNames

template<typename T >
std::vector<std::string> TMVA::Experimental::SOFIE::ROperator_Custom< T >::fInputNames
private

Definition at line 20 of file ROperator_Custom.hxx.

◆ fInputSizes

template<typename T >
std::vector<std::size_t> TMVA::Experimental::SOFIE::ROperator_Custom< T >::fInputSizes
private

Definition at line 23 of file ROperator_Custom.hxx.

◆ fInputType

template<typename T >
ETensorType TMVA::Experimental::SOFIE::ROperator_Custom< T >::fInputType
private

Definition at line 25 of file ROperator_Custom.hxx.

◆ fOpName

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Custom< T >::fOpName
private

Definition at line 19 of file ROperator_Custom.hxx.

◆ fOutputNames

template<typename T >
std::vector<std::string> TMVA::Experimental::SOFIE::ROperator_Custom< T >::fOutputNames
private

Definition at line 21 of file ROperator_Custom.hxx.

◆ fOutputShapes

template<typename T >
std::vector<std::vector<std::size_t> > TMVA::Experimental::SOFIE::ROperator_Custom< T >::fOutputShapes
private

Definition at line 22 of file ROperator_Custom.hxx.

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