Logo ROOT  

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

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

Definition at line 15 of file ROperator_Pad.hxx.

Public Types

enum  EMode { kConstant , kReflect , kEdge , kWrap }
 

Public Member Functions

 ROperator_Pad ()
 
 ROperator_Pad (const std::string &nameX, const std::string &nameP, const std::string &nameCV, const std::string &nameAX, const std::string &nameY, const std::string &mode)
 
std::string Generate (std::string OpName) override
 
void Initialize (RModel &model) override
 
std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > > input) override
 
std::vector< ETensorType > TypeInference (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

T fConstantValue
 
std::vector< size_t > fInputShape
 
EMode fMode
 
std::string fNAX
 
std::string fNCV
 
std::string fNP
 
std::string fNX
 
std::string fNY
 
std::vector< size_t > fOutputShape
 
std::vector< std::pair< int64_t, int64_t > > fPads
 

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)
 
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_Pad.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Pad< T >:
[legend]

Member Enumeration Documentation

◆ EMode

template<typename T >
enum TMVA::Experimental::SOFIE::ROperator_Pad::EMode
Enumerator
kConstant 
kReflect 
kEdge 
kWrap 

Definition at line 18 of file ROperator_Pad.hxx.

Constructor & Destructor Documentation

◆ ROperator_Pad() [1/2]

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

Definition at line 34 of file ROperator_Pad.hxx.

◆ ROperator_Pad() [2/2]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Pad< T >::ROperator_Pad ( const std::string & nameX,
const std::string & nameP,
const std::string & nameCV,
const std::string & nameAX,
const std::string & nameY,
const std::string & mode )
inline

Definition at line 35 of file ROperator_Pad.hxx.

Member Function Documentation

◆ Generate()

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::Generate ( std::string OpName)
inlineoverridevirtual

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

Definition at line 151 of file ROperator_Pad.hxx.

◆ Initialize()

template<typename T >
void TMVA::Experimental::SOFIE::ROperator_Pad< T >::Initialize ( RModel & model)
inlineoverridevirtual

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

Definition at line 64 of file ROperator_Pad.hxx.

◆ ShapeInference()

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

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

Definition at line 59 of file ROperator_Pad.hxx.

◆ TypeInference()

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

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

Definition at line 55 of file ROperator_Pad.hxx.

Member Data Documentation

◆ fConstantValue

template<typename T >
T TMVA::Experimental::SOFIE::ROperator_Pad< T >::fConstantValue
private

Definition at line 26 of file ROperator_Pad.hxx.

◆ fInputShape

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Pad< T >::fInputShape
private

Definition at line 28 of file ROperator_Pad.hxx.

◆ fMode

template<typename T >
EMode TMVA::Experimental::SOFIE::ROperator_Pad< T >::fMode
private

Definition at line 27 of file ROperator_Pad.hxx.

◆ fNAX

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::fNAX
private

Definition at line 24 of file ROperator_Pad.hxx.

◆ fNCV

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::fNCV
private

Definition at line 23 of file ROperator_Pad.hxx.

◆ fNP

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::fNP
private

Definition at line 22 of file ROperator_Pad.hxx.

◆ fNX

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::fNX
private

Definition at line 21 of file ROperator_Pad.hxx.

◆ fNY

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Pad< T >::fNY
private

Definition at line 25 of file ROperator_Pad.hxx.

◆ fOutputShape

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Pad< T >::fOutputShape
private

Definition at line 29 of file ROperator_Pad.hxx.

◆ fPads

template<typename T >
std::vector<std::pair<int64_t, int64_t> > TMVA::Experimental::SOFIE::ROperator_Pad< T >::fPads
private

Definition at line 30 of file ROperator_Pad.hxx.

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