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

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

Definition at line 15 of file ROperator_LayerNormalization.hxx.

Public Member Functions

 ROperator_LayerNormalization ()
 
 ROperator_LayerNormalization (int axis, float epsilon, size_t stashType, const std::string &nameX, const std::string &nameScale, const std::string &nameB, const std::string &nameY, const std::string &nameMean, const std::string &nameInvStdDev)
 
std::string Generate (std::string OpName) override
 
std::string GenerateInitCode () override
 
std::vector< std::string > GetBlasRoutines () override
 
std::vector< std::string > GetStdLibs () override
 
void Initialize (RModel &model) override
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input) override
 
std::vector< ETensorTypeTypeInference (std::vector< ETensorType > input) override
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual ~ROperator ()
 
virtual std::string GenerateSessionMembersCode (std::string)
 
virtual std::string Header ()
 

Private Attributes

int fAttrAxis
 
float fAttrEpsilon
 
size_t fAttrStashType
 
std::string fAxesLength
 
std::vector< DimfAxesShape
 
size_t fAxis
 
std::string fLength
 
std::string fNB
 
std::string fNBroadcastedB
 
std::string fNCastedX
 
std::string fNInvStdDev
 
std::string fNMean
 
std::string fNNormalizedX
 
std::string fNormalizedLength
 
std::vector< DimfNormalizedShape
 
std::string fNScale
 
std::string fNX
 
std::string fNY
 
std::vector< size_t > fShapeB
 
std::vector< DimfShapeInvStdDev
 
std::vector< DimfShapeMean
 
std::vector< DimfShapeScale
 
std::vector< DimfShapeX
 
std::vector< DimfShapeY
 
size_t fSize
 
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_LayerNormalization.hxx>

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

Constructor & Destructor Documentation

◆ ROperator_LayerNormalization() [1/2]

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

Definition at line 53 of file ROperator_LayerNormalization.hxx.

◆ ROperator_LayerNormalization() [2/2]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::ROperator_LayerNormalization ( int  axis,
float  epsilon,
size_t  stashType,
const std::string &  nameX,
const std::string &  nameScale,
const std::string &  nameB,
const std::string &  nameY,
const std::string &  nameMean,
const std::string &  nameInvStdDev 
)
inline

Definition at line 55 of file ROperator_LayerNormalization.hxx.

Member Function Documentation

◆ Generate()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::Generate ( std::string  OpName)
inlineoverridevirtual

◆ GenerateInitCode()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GenerateInitCode ( )
inlineoverridevirtual

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

Definition at line 132 of file ROperator_LayerNormalization.hxx.

◆ GetBlasRoutines()

template<typename T >
std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GetBlasRoutines ( )
inlineoverridevirtual

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

Definition at line 328 of file ROperator_LayerNormalization.hxx.

◆ GetStdLibs()

template<typename T >
std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::GetStdLibs ( )
inlineoverridevirtual

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

Definition at line 330 of file ROperator_LayerNormalization.hxx.

◆ Initialize()

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

◆ ShapeInference()

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

◆ TypeInference()

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

Member Data Documentation

◆ fAttrAxis

template<typename T >
int TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAttrAxis
private

Definition at line 17 of file ROperator_LayerNormalization.hxx.

◆ fAttrEpsilon

template<typename T >
float TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAttrEpsilon
private

Definition at line 18 of file ROperator_LayerNormalization.hxx.

◆ fAttrStashType

template<typename T >
size_t TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAttrStashType
private

Definition at line 19 of file ROperator_LayerNormalization.hxx.

◆ fAxesLength

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAxesLength
private

Definition at line 48 of file ROperator_LayerNormalization.hxx.

◆ fAxesShape

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAxesShape
private

Definition at line 44 of file ROperator_LayerNormalization.hxx.

◆ fAxis

template<typename T >
size_t TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fAxis
private

Definition at line 39 of file ROperator_LayerNormalization.hxx.

◆ fLength

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fLength
private

Definition at line 46 of file ROperator_LayerNormalization.hxx.

◆ fNB

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

Definition at line 23 of file ROperator_LayerNormalization.hxx.

◆ fNBroadcastedB

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNBroadcastedB
private

Definition at line 30 of file ROperator_LayerNormalization.hxx.

◆ fNCastedX

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNCastedX
private

Definition at line 28 of file ROperator_LayerNormalization.hxx.

◆ fNInvStdDev

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNInvStdDev
private

Definition at line 26 of file ROperator_LayerNormalization.hxx.

◆ fNMean

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNMean
private

Definition at line 25 of file ROperator_LayerNormalization.hxx.

◆ fNNormalizedX

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNNormalizedX
private

Definition at line 29 of file ROperator_LayerNormalization.hxx.

◆ fNormalizedLength

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNormalizedLength
private

Definition at line 47 of file ROperator_LayerNormalization.hxx.

◆ fNormalizedShape

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNormalizedShape
private

Definition at line 43 of file ROperator_LayerNormalization.hxx.

◆ fNScale

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNScale
private

Definition at line 22 of file ROperator_LayerNormalization.hxx.

◆ fNX

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fNX
private

Definition at line 21 of file ROperator_LayerNormalization.hxx.

◆ fNY

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

Definition at line 24 of file ROperator_LayerNormalization.hxx.

◆ fShapeB

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fShapeB
private

Definition at line 34 of file ROperator_LayerNormalization.hxx.

◆ fShapeInvStdDev

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fShapeInvStdDev
private

Definition at line 37 of file ROperator_LayerNormalization.hxx.

◆ fShapeMean

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fShapeMean
private

Definition at line 36 of file ROperator_LayerNormalization.hxx.

◆ fShapeScale

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fShapeScale
private

Definition at line 33 of file ROperator_LayerNormalization.hxx.

◆ fShapeX

template<typename T >
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fShapeX
private

Definition at line 32 of file ROperator_LayerNormalization.hxx.

◆ fShapeY

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

Definition at line 35 of file ROperator_LayerNormalization.hxx.

◆ fSize

template<typename T >
size_t TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >::fSize
private

Definition at line 40 of file ROperator_LayerNormalization.hxx.

◆ fType

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

Definition at line 50 of file ROperator_LayerNormalization.hxx.

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