Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
List of all members | Public Member Functions | Protected Attributes | Private Member Functions | Private Attributes | List of all members
TMVA::Experimental::SOFIE::ROperator_ScatterElements Class Referencefinal

Definition at line 15 of file ROperator_ScatterElements.hxx.

Public Member Functions

 ROperator_ScatterElements ()
 
 ROperator_ScatterElements (const std::string &nameX, const std::string &nameI, const std::string &nameU, const std::string &nameY, int axis, std::string reduction)
 
std::string Generate (std::string opName) override
 
virtual std::string GenerateDeclCode ()
 
std::string GenerateInitCode () override
 
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 ()
 
void Initialize (RModel &model) override
 
bool IsOutputConstant () const
 check if the output of the operator is Constant and is evaluated at initialization time
 

Protected Attributes

std::vector< std::string_view > fInputTensorNames
 
bool fIsOutputConstant = false
 flag to identify if operator has a constant output (no need to generate code)
 
bool fIsOutputParamShape = false
 flag to identify of the output represents a parametric shape (can be known at compile time)
 
std::vector< std::string_view > fOutputTensorNames
 
const std::string SP = " "
 space used to correctly indent the generated C++ code
 

Private Member Functions

std::string ReductionFunction (const std::string &t1, const std::string &t2)
 

Private Attributes

int64_t fAxis
 
std::string fNI
 
std::string fNU
 
std::string fNX
 
std::string fNY
 
std::string fReduction
 
std::vector< Dim > fShapeI
 
std::vector< Dim > fShapeX
 
std::vector< Dim > fShapeY
 

#include <TMVA/ROperator_ScatterElements.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_ScatterElements:
[legend]

Constructor & Destructor Documentation

◆ ROperator_ScatterElements() [1/2]

TMVA::Experimental::SOFIE::ROperator_ScatterElements::ROperator_ScatterElements ( )
inline

Definition at line 51 of file ROperator_ScatterElements.hxx.

◆ ROperator_ScatterElements() [2/2]

TMVA::Experimental::SOFIE::ROperator_ScatterElements::ROperator_ScatterElements ( const std::string & nameX,
const std::string & nameI,
const std::string & nameU,
const std::string & nameY,
int axis,
std::string reduction )
inline

Definition at line 52 of file ROperator_ScatterElements.hxx.

Member Function Documentation

◆ Generate()

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::Generate ( std::string opName)
inlineoverridevirtual

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

Definition at line 108 of file ROperator_ScatterElements.hxx.

◆ GenerateDeclCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateDeclCode ( )
inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_Random.

Definition at line 26 of file ROperator.hxx.

◆ GenerateInitCode()

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::GenerateInitCode ( )
inlineoverridevirtual

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

Definition at line 103 of file ROperator_ScatterElements.hxx.

◆ GenerateSessionMembersCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateSessionMembersCode ( std::string )
inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_LSTM< T >, TMVA::Experimental::SOFIE::ROperator_RNN< T >, and TMVA::Experimental::SOFIE::ROperator_NonZero< T >.

Definition at line 28 of file ROperator.hxx.

◆ GetBlasRoutines()

virtual std::vector< std::string > TMVA::Experimental::SOFIE::ROperator::GetBlasRoutines ( )
inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_BatchNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Conv< T >, TMVA::Experimental::SOFIE::ROperator_ConvTranspose< T >, TMVA::Experimental::SOFIE::ROperator_Einsum< T >, TMVA::Experimental::SOFIE::ROperator_Gemm< T >, TMVA::Experimental::SOFIE::ROperator_GRU< T >, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_LSTM< T >, and TMVA::Experimental::SOFIE::ROperator_RNN< T >.

Definition at line 19 of file ROperator.hxx.

◆ GetOpInputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpInputTensors ( ) const
inlineinherited

Definition at line 48 of file ROperator.hxx.

◆ GetOpOutputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpOutputTensors ( ) const
inlineinherited

Definition at line 52 of file ROperator.hxx.

◆ GetStdLibs()

virtual std::vector< std::string > TMVA::Experimental::SOFIE::ROperator::GetStdLibs ( )
inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_Basic_Is< Op >, TMVA::Experimental::SOFIE::ROperator_BasicBinary< T, Op >, TMVA::Experimental::SOFIE::ROperator_BasicNary< T, Op >, TMVA::Experimental::SOFIE::ROperator_BasicUnary< T, Op >, TMVA::Experimental::SOFIE::ROperator_Erf< T >, TMVA::Experimental::SOFIE::ROperator_Gelu, TMVA::Experimental::SOFIE::ROperator_HardSigmoid< T >, TMVA::Experimental::SOFIE::ROperator_HardSwish< T >, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Random, TMVA::Experimental::SOFIE::ROperator_Selu< T >, TMVA::Experimental::SOFIE::ROperator_Sigmoid< T >, TMVA::Experimental::SOFIE::ROperator_Softmax, TMVA::Experimental::SOFIE::ROperator_Swish, and TMVA::Experimental::SOFIE::ROperator_Tanh< T >.

Definition at line 20 of file ROperator.hxx.

◆ Header()

virtual std::string TMVA::Experimental::SOFIE::ROperator::Header ( )
inlinevirtualinherited

Definition at line 29 of file ROperator.hxx.

◆ Initialize()

void TMVA::Experimental::SOFIE::ROperator_ScatterElements::Initialize ( RModel & model)
inlineoverridevirtual

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

Definition at line 64 of file ROperator_ScatterElements.hxx.

◆ IsOutputConstant()

bool TMVA::Experimental::SOFIE::ROperator::IsOutputConstant ( ) const
inlineinherited

check if the output of the operator is Constant and is evaluated at initialization time

Definition at line 32 of file ROperator.hxx.

◆ ReductionFunction()

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::ReductionFunction ( const std::string & t1,
const std::string & t2 )
inlineprivate

Definition at line 32 of file ROperator_ScatterElements.hxx.

Member Data Documentation

◆ fAxis

int64_t TMVA::Experimental::SOFIE::ROperator_ScatterElements::fAxis
private

Definition at line 18 of file ROperator_ScatterElements.hxx.

◆ fInputTensorNames

std::vector<std::string_view> TMVA::Experimental::SOFIE::ROperator::fInputTensorNames
mutableprotectedinherited

Definition at line 44 of file ROperator.hxx.

◆ fIsOutputConstant

bool TMVA::Experimental::SOFIE::ROperator::fIsOutputConstant = false
protectedinherited

flag to identify if operator has a constant output (no need to generate code)

Definition at line 41 of file ROperator.hxx.

◆ fIsOutputParamShape

bool TMVA::Experimental::SOFIE::ROperator::fIsOutputParamShape = false
protectedinherited

flag to identify of the output represents a parametric shape (can be known at compile time)

Definition at line 42 of file ROperator.hxx.

◆ fNI

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::fNI
private

Definition at line 21 of file ROperator_ScatterElements.hxx.

◆ fNU

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::fNU
private

Definition at line 22 of file ROperator_ScatterElements.hxx.

◆ fNX

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::fNX
private

Definition at line 20 of file ROperator_ScatterElements.hxx.

◆ fNY

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::fNY
private

Definition at line 23 of file ROperator_ScatterElements.hxx.

◆ fOutputTensorNames

std::vector<std::string_view> TMVA::Experimental::SOFIE::ROperator::fOutputTensorNames
mutableprotectedinherited

Definition at line 45 of file ROperator.hxx.

◆ fReduction

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::fReduction
private

Definition at line 24 of file ROperator_ScatterElements.hxx.

◆ fShapeI

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_ScatterElements::fShapeI
private

Definition at line 27 of file ROperator_ScatterElements.hxx.

◆ fShapeX

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_ScatterElements::fShapeX
private

Definition at line 26 of file ROperator_ScatterElements.hxx.

◆ fShapeY

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_ScatterElements::fShapeY
private

Definition at line 28 of file ROperator_ScatterElements.hxx.

◆ SP

const std::string TMVA::Experimental::SOFIE::ROperator::SP = " "
protectedinherited

space used to correctly indent the generated C++ code

Definition at line 40 of file ROperator.hxx.

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