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, EActivationType activation=EActivationType::UNDEFINED)

	ROperator_Gemm (float alpha, float beta, int_t transA, int_t transB, std::string nameA, std::string nameB, std::string nameY, EActivationType activation=EActivationType::UNDEFINED)

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) override

std::string	GenerateInitCode () override

std::vector< std::string >	GetBlasRoutines () 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	GenerateSessionMembersCode (std::string)

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
EActivationType	fActivation

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< Dim >	fShapeA

std::vector< Dim >	fShapeB

std::vector< size_t >	fShapeC

std::vector< Dim >	fShapeY

std::string	fType

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_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 47 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,
		EActivationType	activation = EActivationType::UNDEFINED )

inline

Definition at line 48 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,
		EActivationType	activation = EActivationType::UNDEFINED )

inline

Definition at line 60 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 76 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 134 of file ROperator_Gemm.hxx.

◆ Generate()

template<typename T >

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

inlineoverridevirtual

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

Definition at line 293 of file ROperator_Gemm.hxx.

◆ GenerateInitCode()

template<typename T >

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

inlineoverridevirtual

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

Definition at line 272 of file ROperator_Gemm.hxx.

◆ GetBlasRoutines()

template<typename T >

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

inlineoverridevirtual

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

Definition at line 385 of file ROperator_Gemm.hxx.

◆ Initialize()

template<typename T >

void TMVA::Experimental::SOFIE::ROperator_Gemm< T >::Initialize ( RModel & model )

inlineoverridevirtual

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

Definition at line 140 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 )

inlineoverridevirtual

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

Definition at line 131 of file ROperator_Gemm.hxx.

◆ TypeInference()

template<typename T >

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

inlineoverridevirtual

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

Definition at line 70 of file ROperator_Gemm.hxx.

Member Data Documentation

◆ fActivation

template<typename T >

EActivationType TMVA::Experimental::SOFIE::ROperator_Gemm< T >::fActivation

private

Definition at line 39 of file ROperator_Gemm.hxx.

◆ 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 40 of file ROperator_Gemm.hxx.

◆ fShapeB

template<typename T >

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

private

Definition at line 41 of file ROperator_Gemm.hxx.

◆ fShapeC

template<typename T >

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

private

Definition at line 42 of file ROperator_Gemm.hxx.

◆ fShapeY

template<typename T >

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

private

Definition at line 43 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

Public Member Functions

Private Attributes

Additional Inherited Members

Constructor & Destructor Documentation

◆ ROperator_Gemm() [1/3]

◆ ROperator_Gemm() [2/3]

◆ ROperator_Gemm() [3/3]

Member Function Documentation

◆ DoShapeInference()

◆ DynamicShapeInference()

◆ Generate()

◆ GenerateInitCode()

◆ GetBlasRoutines()

◆ Initialize()

◆ ShapeInference()

◆ TypeInference()

Member Data Documentation

◆ fActivation

◆ fAttrAlpha

◆ fAttrBeta

◆ fAttrTransA

◆ fAttrTransB

◆ fIsDynamic

◆ fNA

◆ fNB

◆ fNC

◆ fNC2

◆ fNY

◆ fShapeA

◆ fShapeB

◆ fShapeC

◆ fShapeY

◆ fType