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

Definition at line 20 of file ROperator_Reshape.hxx.

Public Member Functions

 ROperator_Reshape ()
 
 ROperator_Reshape (ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameInput2, std::string nameOutput)
 
 ROperator_Reshape (ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput)
 
std::string Generate (std::string opName) override
 
void Initialize (RModel &model) override
 
std::string Name () const
 
std::vector< std::vector< Dim > > ShapeInference (const std::vector< std::vector< Dim > > &input)
 
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 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 ()
 

Private Attributes

int fAllowZero = 0
 
std::vector< int64_t > fAttrAxes
 
int fAxis = 1
 
bool fDimInput = false
 
bool fDynamicShape = false
 
std::string fNData
 
std::string fNInput2
 
std::string fNOutput
 
ReshapeOpMode fOpMode = Reshape
 
std::vector< int64_t > fShape
 
std::vector< DimfShapeInput
 
std::vector< DimfShapeOutput
 
bool fVerbose = false
 

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_Reshape.hxx>

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

Constructor & Destructor Documentation

◆ ROperator_Reshape() [1/3]

TMVA::Experimental::SOFIE::ROperator_Reshape::ROperator_Reshape ( )
inline

Definition at line 51 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [2/3]

TMVA::Experimental::SOFIE::ROperator_Reshape::ROperator_Reshape ( ReshapeOpMode opMode,
int attr_value,
std::string nameData,
std::string nameInput2,
std::string nameOutput )
inline

Definition at line 52 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [3/3]

TMVA::Experimental::SOFIE::ROperator_Reshape::ROperator_Reshape ( ReshapeOpMode opMode,
std::vector< int64_t > attrAxes,
std::string nameData,
std::string nameOutput )
inline

Definition at line 68 of file ROperator_Reshape.hxx.

Member Function Documentation

◆ Generate()

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

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

Definition at line 316 of file ROperator_Reshape.hxx.

◆ Initialize()

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

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

Definition at line 233 of file ROperator_Reshape.hxx.

◆ Name()

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::Name ( ) const
inline

Definition at line 43 of file ROperator_Reshape.hxx.

◆ ShapeInference() [1/2]

std::vector< std::vector< Dim > > TMVA::Experimental::SOFIE::ROperator_Reshape::ShapeInference ( const std::vector< std::vector< Dim > > & input)
inline

Definition at line 85 of file ROperator_Reshape.hxx.

◆ ShapeInference() [2/2]

std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_Reshape::ShapeInference ( std::vector< std::vector< size_t > > input)
inlineoverridevirtual

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

Definition at line 80 of file ROperator_Reshape.hxx.

◆ TypeInference()

std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_Reshape::TypeInference ( std::vector< ETensorType > input)
inlineoverridevirtual

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

Definition at line 76 of file ROperator_Reshape.hxx.

Member Data Documentation

◆ fAllowZero

int TMVA::Experimental::SOFIE::ROperator_Reshape::fAllowZero = 0
private

Definition at line 30 of file ROperator_Reshape.hxx.

◆ fAttrAxes

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reshape::fAttrAxes
private

Definition at line 38 of file ROperator_Reshape.hxx.

◆ fAxis

int TMVA::Experimental::SOFIE::ROperator_Reshape::fAxis = 1
private

Definition at line 31 of file ROperator_Reshape.hxx.

◆ fDimInput

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fDimInput = false
private

Definition at line 26 of file ROperator_Reshape.hxx.

◆ fDynamicShape

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fDynamicShape = false
private

Definition at line 27 of file ROperator_Reshape.hxx.

◆ fNData

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNData
private

Definition at line 33 of file ROperator_Reshape.hxx.

◆ fNInput2

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNInput2
private

Definition at line 34 of file ROperator_Reshape.hxx.

◆ fNOutput

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNOutput
private

Definition at line 35 of file ROperator_Reshape.hxx.

◆ fOpMode

ReshapeOpMode TMVA::Experimental::SOFIE::ROperator_Reshape::fOpMode = Reshape
private

Definition at line 28 of file ROperator_Reshape.hxx.

◆ fShape

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reshape::fShape
private

Definition at line 39 of file ROperator_Reshape.hxx.

◆ fShapeInput

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reshape::fShapeInput
private

Definition at line 36 of file ROperator_Reshape.hxx.

◆ fShapeOutput

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reshape::fShapeOutput
private

Definition at line 37 of file ROperator_Reshape.hxx.

◆ fVerbose

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fVerbose = false
private

Definition at line 25 of file ROperator_Reshape.hxx.

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