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

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

Definition at line 17 of file ROperator_Reshape.hxx.

Public Member Functions

 ROperator_Reshape ()
 
 ROperator_Reshape (ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameShape, std::string nameOutput)
 
 ROperator_Reshape (ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput)
 
std::string Generate (std::string OpName)
 
void Initialize (RModel &model)
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input)
 
std::vector< ETensorTypeTypeInference (std::vector< ETensorType > input)
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual ~ROperator ()
 
virtual std::string GenerateInitCode ()
 
virtual std::string GenerateSessionMembersCode (std::string)
 
virtual std::string Header ()
 

Private Attributes

int fAllowZero = 0
 
std::vector< int64_t > fAttrAxes
 
int fAxis = 1
 
std::string fNData
 
std::string fNOutput
 
std::string fNShape
 
ReshapeOpMode fOpMode = Reshape
 
std::vector< size_t > fShapeInput
 
std::vector< size_t > fShapeOutput
 

Additional Inherited Members

- Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator
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< T >:
[legend]

Constructor & Destructor Documentation

◆ ROperator_Reshape() [1/3]

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

Definition at line 36 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [2/3]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Reshape< T >::ROperator_Reshape ( ReshapeOpMode  opMode,
int  attr_value,
std::string  nameData,
std::string  nameShape,
std::string  nameOutput 
)
inline

Definition at line 37 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [3/3]

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

Definition at line 47 of file ROperator_Reshape.hxx.

Member Function Documentation

◆ Generate()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Reshape< T >::Generate ( std::string  OpName)
inlinevirtual

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

Definition at line 177 of file ROperator_Reshape.hxx.

◆ Initialize()

template<typename T >
void TMVA::Experimental::SOFIE::ROperator_Reshape< T >::Initialize ( RModel model)
inlinevirtual

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

Definition at line 140 of file ROperator_Reshape.hxx.

◆ ShapeInference()

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

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

Definition at line 61 of file ROperator_Reshape.hxx.

◆ TypeInference()

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

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

Definition at line 55 of file ROperator_Reshape.hxx.

Member Data Documentation

◆ fAllowZero

template<typename T >
int TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fAllowZero = 0
private

Definition at line 24 of file ROperator_Reshape.hxx.

◆ fAttrAxes

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

Definition at line 32 of file ROperator_Reshape.hxx.

◆ fAxis

template<typename T >
int TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fAxis = 1
private

Definition at line 25 of file ROperator_Reshape.hxx.

◆ fNData

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fNData
private

Definition at line 27 of file ROperator_Reshape.hxx.

◆ fNOutput

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fNOutput
private

Definition at line 29 of file ROperator_Reshape.hxx.

◆ fNShape

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fNShape
private

Definition at line 28 of file ROperator_Reshape.hxx.

◆ fOpMode

template<typename T >
ReshapeOpMode TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fOpMode = Reshape
private

Definition at line 22 of file ROperator_Reshape.hxx.

◆ fShapeInput

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fShapeInput
private

Definition at line 30 of file ROperator_Reshape.hxx.

◆ fShapeOutput

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Reshape< T >::fShapeOutput
private

Definition at line 31 of file ROperator_Reshape.hxx.

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