Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::ROperator_ScatterND Class Referencefinal

Definition at line 16 of file ROperator_ScatterND.hxx.

Public Member Functions

 ROperator_ScatterND ()
 
 ROperator_ScatterND (const std::string &nameX, const std::string &nameI, const std::string &nameU, const std::string &nameY, std::string reduction)
 
std::string Generate (std::string opName) override
 
void Initialize (RModel &model) override
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual ~ROperator ()
 
virtual std::string GenerateDeclCode ()
 
virtual std::string GenerateInitCode ()
 
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 ()
 
virtual std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > >)
 
virtual std::vector< ETensorTypeTypeInference (std::vector< ETensorType >)
 

Private Attributes

std::vector< int64_t > fIndices
 
std::string fNI
 
std::string fNU
 
std::string fNX
 
std::string fNY
 
std::string fReduction
 
std::vector< DimfShapeI
 
std::vector< DimfShapeX
 
std::vector< DimfShapeY
 
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)
 
bool fIsOutputParamShape = false
 flag to identify of the output represents a parametric shape (can be knwon at compile time)
 
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_ScatterND.hxx>

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

Constructor & Destructor Documentation

◆ ROperator_ScatterND() [1/2]

TMVA::Experimental::SOFIE::ROperator_ScatterND::ROperator_ScatterND ( )
inline

Definition at line 38 of file ROperator_ScatterND.hxx.

◆ ROperator_ScatterND() [2/2]

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

Definition at line 39 of file ROperator_ScatterND.hxx.

Member Function Documentation

◆ Generate()

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

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

Definition at line 100 of file ROperator_ScatterND.hxx.

◆ Initialize()

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

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

Definition at line 48 of file ROperator_ScatterND.hxx.

Member Data Documentation

◆ fIndices

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_ScatterND::fIndices
private

Definition at line 32 of file ROperator_ScatterND.hxx.

◆ fNI

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

Definition at line 22 of file ROperator_ScatterND.hxx.

◆ fNU

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

Definition at line 23 of file ROperator_ScatterND.hxx.

◆ fNX

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

Definition at line 21 of file ROperator_ScatterND.hxx.

◆ fNY

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

Definition at line 24 of file ROperator_ScatterND.hxx.

◆ fReduction

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

Definition at line 25 of file ROperator_ScatterND.hxx.

◆ fShapeI

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

Definition at line 28 of file ROperator_ScatterND.hxx.

◆ fShapeX

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

Definition at line 27 of file ROperator_ScatterND.hxx.

◆ fShapeY

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

Definition at line 29 of file ROperator_ScatterND.hxx.

◆ fType

std::string TMVA::Experimental::SOFIE::ROperator_ScatterND::fType
private

Definition at line 34 of file ROperator_ScatterND.hxx.

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