Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::ROperator_Gemm< T > Class Template Referencefinal

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

Definition at line 22 of file ROperator_Gemm.hxx.

Public Member Functions

 ROperator_Gemm ()
 
 ROperator_Gemm (float alpha, float beta, int_t transA, int_t transB, std::string nameA, std::string nameB, std::string nameC, std::string nameY)
 
 ROperator_Gemm (float alpha, float beta, int_t transA, int_t transB, std::string nameA, std::string nameB, std::string nameY)
 
template<typename U >
std::vector< std::vector< U > > DoShapeInference (const std::vector< std::vector< U > > &input)
 
std::vector< std::vector< Dim > > DynamicShapeInference (const std::vector< std::vector< Dim > > &input)
 
std::string Generate (std::string opName)
 
std::string GenerateInitCode ()
 
std::vector< std::string > GetBlasRoutines ()
 
void Initialize (RModel &model)
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input)
 
std::vector< ETensorTypeTypeInference (std::vector< ETensorType > input)
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual ~ROperator ()
 
virtual std::string GenerateSessionMembersCode (std::string)
 
virtual std::vector< std::string > GetStdLibs ()
 
virtual std::string Header ()
 

Private Attributes

float fAttrAlpha = 1.0
 
float fAttrBeta = 1.0
 
int_t fAttrTransA = 0
 
int_t fAttrTransB = 0
 
bool fIsDynamic = false
 
std::string fNA
 
std::string fNB
 
std::string fNC = ""
 
std::string fNC2
 
std::string fNY
 
std::vector< DimfShapeA
 
std::vector< DimfShapeB
 
std::vector< size_t > fShapeC
 
std::vector< DimfShapeY
 
std::string fType
 

Additional Inherited Members

- Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator
bool fIsOutputConstant = false
 flag to identify if operator has a constant output (no need to generate code)
 
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_Gemm.hxx>

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

Constructor & Destructor Documentation

◆ ROperator_Gemm() [1/3]

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

Definition at line 46 of file ROperator_Gemm.hxx.

◆ ROperator_Gemm() [2/3]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Gemm< T >::ROperator_Gemm ( float  alpha,
float  beta,
int_t  transA,
int_t  transB,
std::string  nameA,
std::string  nameB,
std::string  nameY 
)
inline

Definition at line 47 of file ROperator_Gemm.hxx.

◆ ROperator_Gemm() [3/3]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Gemm< T >::ROperator_Gemm ( float  alpha,
float  beta,
int_t  transA,
int_t  transB,
std::string  nameA,
std::string  nameB,
std::string  nameC,
std::string  nameY 
)
inline

Definition at line 56 of file ROperator_Gemm.hxx.

Member Function Documentation

◆ DoShapeInference()

template<typename T >
template<typename U >
std::vector< std::vector< U > > TMVA::Experimental::SOFIE::ROperator_Gemm< T >::DoShapeInference ( const std::vector< std::vector< U > > &  input)
inline

Definition at line 71 of file ROperator_Gemm.hxx.

◆ DynamicShapeInference()

template<typename T >
std::vector< std::vector< Dim > > TMVA::Experimental::SOFIE::ROperator_Gemm< T >::DynamicShapeInference ( const std::vector< std::vector< Dim > > &  input)
inline

Definition at line 116 of file ROperator_Gemm.hxx.

◆ Generate()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::Generate ( std::string  opName)
inlinevirtual

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

Definition at line 263 of file ROperator_Gemm.hxx.

◆ GenerateInitCode()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::GenerateInitCode ( )
inlinevirtual

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

Definition at line 241 of file ROperator_Gemm.hxx.

◆ GetBlasRoutines()

template<typename T >
std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_Gemm< T >::GetBlasRoutines ( )
inlinevirtual

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

Definition at line 356 of file ROperator_Gemm.hxx.

◆ Initialize()

template<typename T >
void TMVA::Experimental::SOFIE::ROperator_Gemm< T >::Initialize ( RModel model)
inlinevirtual

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

Definition at line 122 of file ROperator_Gemm.hxx.

◆ ShapeInference()

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

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

Definition at line 113 of file ROperator_Gemm.hxx.

◆ TypeInference()

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

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

Definition at line 65 of file ROperator_Gemm.hxx.

Member Data Documentation

◆ fAttrAlpha

template<typename T >
float TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fAttrAlpha = 1.0
private

Definition at line 28 of file ROperator_Gemm.hxx.

◆ fAttrBeta

template<typename T >
float TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fAttrBeta = 1.0
private

Definition at line 29 of file ROperator_Gemm.hxx.

◆ fAttrTransA

template<typename T >
int_t TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fAttrTransA = 0
private

Definition at line 30 of file ROperator_Gemm.hxx.

◆ fAttrTransB

template<typename T >
int_t TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fAttrTransB = 0
private

Definition at line 31 of file ROperator_Gemm.hxx.

◆ fIsDynamic

template<typename T >
bool TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fIsDynamic = false
private

Definition at line 26 of file ROperator_Gemm.hxx.

◆ fNA

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fNA
private

Definition at line 33 of file ROperator_Gemm.hxx.

◆ fNB

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fNB
private

Definition at line 34 of file ROperator_Gemm.hxx.

◆ fNC

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fNC = ""
private

Definition at line 35 of file ROperator_Gemm.hxx.

◆ fNC2

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fNC2
private

Definition at line 36 of file ROperator_Gemm.hxx.

◆ fNY

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fNY
private

Definition at line 37 of file ROperator_Gemm.hxx.

◆ fShapeA

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fShapeA
private

Definition at line 39 of file ROperator_Gemm.hxx.

◆ fShapeB

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fShapeB
private

Definition at line 40 of file ROperator_Gemm.hxx.

◆ fShapeC

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fShapeC
private

Definition at line 41 of file ROperator_Gemm.hxx.

◆ fShapeY

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fShapeY
private

Definition at line 42 of file ROperator_Gemm.hxx.

◆ fType

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fType
private

Definition at line 38 of file ROperator_Gemm.hxx.

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