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

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

Definition at line 15 of file ROperator_Softmax.hxx.

Public Member Functions

 ROperator_Softmax ()
 
 ROperator_Softmax (int64_t attr_axis, std::string nameX, std::string nameY)
 
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::vector< std::string > GetBlasRoutines ()
 
virtual std::vector< std::string > GetStdLibs ()
 
virtual std::string Header ()
 

Private Attributes

int64_t fAttrAxis
 
std::string fNX
 
std::string fNY
 
std::vector< size_t > fShape
 
std::string fType
 

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

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

Constructor & Destructor Documentation

◆ ROperator_Softmax() [1/2]

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

Definition at line 27 of file ROperator_Softmax.hxx.

◆ ROperator_Softmax() [2/2]

template<typename T >
TMVA::Experimental::SOFIE::ROperator_Softmax< T >::ROperator_Softmax ( int64_t  attr_axis,
std::string  nameX,
std::string  nameY 
)
inline

Definition at line 28 of file ROperator_Softmax.hxx.

Member Function Documentation

◆ Generate()

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

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

Definition at line 52 of file ROperator_Softmax.hxx.

◆ Initialize()

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

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

Definition at line 41 of file ROperator_Softmax.hxx.

◆ ShapeInference()

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

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

Definition at line 35 of file ROperator_Softmax.hxx.

◆ TypeInference()

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

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

Definition at line 33 of file ROperator_Softmax.hxx.

Member Data Documentation

◆ fAttrAxis

template<typename T >
int64_t TMVA::Experimental::SOFIE::ROperator_Softmax< T >::fAttrAxis
private

Definition at line 18 of file ROperator_Softmax.hxx.

◆ fNX

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

Definition at line 20 of file ROperator_Softmax.hxx.

◆ fNY

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

Definition at line 21 of file ROperator_Softmax.hxx.

◆ fShape

template<typename T >
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Softmax< T >::fShape
private

Definition at line 22 of file ROperator_Softmax.hxx.

◆ fType

template<typename T >
std::string TMVA::Experimental::SOFIE::ROperator_Softmax< T >::fType
private

Definition at line 24 of file ROperator_Softmax.hxx.

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