Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
List of all members | Public Member Functions | Private Attributes | List of all members
TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op > Class Template Referencefinal

template<typename T, EReduceOpMode Op>
class TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >

Definition at line 22 of file ROperator_Reduce.hxx.

Public Member Functions

 ROperator_Reduce ()
 
 ROperator_Reduce (int keepdims, std::vector< int64_t > attrAxes, std::string nameX, std::string nameAxes, std::string nameY)
 
std::string Generate (std::string opName)
 
void Initialize (RModel &model)
 
std::string Name ()
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input)
 
std::vector< ETensorType > TypeInference (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::vector< std::string > GetBlasRoutines ()
 
virtual std::vector< std::string > GetStdLibs ()
 
virtual std::string Header ()
 

Private Attributes

std::vector< int64_t > fAttrAxes
 
int fkeepdims = 1
 
std::string fNAxes
 
std::string fNX
 
std::string fNY
 
EReduceOpMode fReduceOpMode
 
std::vector< size_t > fShapeX
 
std::vector< size_t > fShapeY
 
std::vector< size_t > fShapeYNotPruned
 

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

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >:
[legend]

Constructor & Destructor Documentation

◆ ROperator_Reduce() [1/2]

template<typename T , EReduceOpMode Op>
TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::ROperator_Reduce ( )
inline

Definition at line 47 of file ROperator_Reduce.hxx.

◆ ROperator_Reduce() [2/2]

template<typename T , EReduceOpMode Op>
TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::ROperator_Reduce ( int  keepdims,
std::vector< int64_t >  attrAxes,
std::string  nameX,
std::string  nameAxes,
std::string  nameY 
)
inline

Definition at line 48 of file ROperator_Reduce.hxx.

Member Function Documentation

◆ Generate()

template<typename T , EReduceOpMode Op>
std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::Generate ( std::string  opName)
inlinevirtual

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

Definition at line 115 of file ROperator_Reduce.hxx.

◆ Initialize()

template<typename T , EReduceOpMode Op>
void TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::Initialize ( RModel &  model)
inlinevirtual

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

Definition at line 85 of file ROperator_Reduce.hxx.

◆ Name()

template<typename T , EReduceOpMode Op>
std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::Name ( )
inline

Definition at line 39 of file ROperator_Reduce.hxx.

◆ ShapeInference()

template<typename T , EReduceOpMode Op>
std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::ShapeInference ( std::vector< std::vector< size_t > >  input)
inlinevirtual

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

Definition at line 59 of file ROperator_Reduce.hxx.

◆ TypeInference()

template<typename T , EReduceOpMode Op>
std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::TypeInference ( std::vector< ETensorType >  input)
inlinevirtual

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

Definition at line 54 of file ROperator_Reduce.hxx.

Member Data Documentation

◆ fAttrAxes

template<typename T , EReduceOpMode Op>
std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fAttrAxes
private

Definition at line 27 of file ROperator_Reduce.hxx.

◆ fkeepdims

template<typename T , EReduceOpMode Op>
int TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fkeepdims = 1
private

Definition at line 26 of file ROperator_Reduce.hxx.

◆ fNAxes

template<typename T , EReduceOpMode Op>
std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNAxes
private

Definition at line 30 of file ROperator_Reduce.hxx.

◆ fNX

template<typename T , EReduceOpMode Op>
std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNX
private

Definition at line 29 of file ROperator_Reduce.hxx.

◆ fNY

template<typename T , EReduceOpMode Op>
std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNY
private

Definition at line 31 of file ROperator_Reduce.hxx.

◆ fReduceOpMode

template<typename T , EReduceOpMode Op>
EReduceOpMode TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fReduceOpMode
private

Definition at line 28 of file ROperator_Reduce.hxx.

◆ fShapeX

template<typename T , EReduceOpMode Op>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeX
private

Definition at line 32 of file ROperator_Reduce.hxx.

◆ fShapeY

template<typename T , EReduceOpMode Op>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeY
private

Definition at line 33 of file ROperator_Reduce.hxx.

◆ fShapeYNotPruned

template<typename T , EReduceOpMode Op>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeYNotPruned
private

Definition at line 34 of file ROperator_Reduce.hxx.

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