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< ETensorType > | TypeInference (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< Dim > | fAxesShape |
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< Dim > | fNormalizedShape |
std::string | fNScale |
std::string | fNX |
std::string | fNY |
std::vector< size_t > | fShapeB |
std::vector< Dim > | fShapeInvStdDev |
std::vector< Dim > | fShapeMean |
std::vector< Dim > | fShapeScale |
std::vector< Dim > | fShapeX |
std::vector< Dim > | fShapeY |
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>
|
inline |
Definition at line 53 of file ROperator_LayerNormalization.hxx.
|
inline |
Definition at line 55 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 147 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 132 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 328 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 330 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 68 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 64 of file ROperator_LayerNormalization.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 66 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 17 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 18 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 19 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 48 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 44 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 39 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 46 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 23 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 30 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 28 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 26 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 25 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 29 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 47 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 43 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 22 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 21 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 24 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 34 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 37 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 36 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 33 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 32 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 35 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 40 of file ROperator_LayerNormalization.hxx.
|
private |
Definition at line 50 of file ROperator_LayerNormalization.hxx.