TMVA::Experimental::SOFIE::ROperator_Softmax Class Reference

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, bool logSoftmax=false)
std::string	Generate (std::string opName) override
std::vector< std::string >	GetStdLibs () 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::string	Header ()
bool	IsOutputConstant () const
	check if the output of the operator is Constant and is evaluated at initialization time

Private Attributes
int64_t	fAttrAxis
bool	fLogSoftmax
std::string	fNX
std::string	fNY
std::vector< Dim >	fShape
std::string	fType
bool	fUseVDT = 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 known 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_Softmax.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Softmax:

Constructor & Destructor Documentation

ROperator_Softmax() [1/2]

TMVA::Experimental::SOFIE::ROperator_Softmax::ROperator_Softmax ( )

inline

Definition at line 29 of file ROperator_Softmax.hxx.

ROperator_Softmax() [2/2]

TMVA::Experimental::SOFIE::ROperator_Softmax::ROperator_Softmax ( int64_t attr_axis, std::string nameX, std::string nameY, bool logSoftmax = false )

inline

Definition at line 30 of file ROperator_Softmax.hxx.

Member Function Documentation

Generate()

std::string TMVA::Experimental::SOFIE::ROperator_Softmax::Generate ( std::string opName )

inlineoverridevirtual

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

Definition at line 65 of file ROperator_Softmax.hxx.

GetStdLibs()

std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_Softmax::GetStdLibs ( )

inlineoverridevirtual

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

Definition at line 191 of file ROperator_Softmax.hxx.

Initialize()

void TMVA::Experimental::SOFIE::ROperator_Softmax::Initialize ( RModel & model )

inlineoverridevirtual

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

Definition at line 46 of file ROperator_Softmax.hxx.

ShapeInference()

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

inlineoverridevirtual

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

Definition at line 41 of file ROperator_Softmax.hxx.

TypeInference()

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

inlineoverridevirtual

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

Definition at line 39 of file ROperator_Softmax.hxx.

Member Data Documentation

fAttrAxis

int64_t TMVA::Experimental::SOFIE::ROperator_Softmax::fAttrAxis

private

Definition at line 20 of file ROperator_Softmax.hxx.

fLogSoftmax

bool TMVA::Experimental::SOFIE::ROperator_Softmax::fLogSoftmax

private

Definition at line 18 of file ROperator_Softmax.hxx.

fNX

std::string TMVA::Experimental::SOFIE::ROperator_Softmax::fNX

private

Definition at line 22 of file ROperator_Softmax.hxx.

fNY

std::string TMVA::Experimental::SOFIE::ROperator_Softmax::fNY

private

Definition at line 23 of file ROperator_Softmax.hxx.

fShape

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Softmax::fShape

private

Definition at line 24 of file ROperator_Softmax.hxx.

fType

std::string TMVA::Experimental::SOFIE::ROperator_Softmax::fType

private

Definition at line 26 of file ROperator_Softmax.hxx.

fUseVDT

bool TMVA::Experimental::SOFIE::ROperator_Softmax::fUseVDT = false

private

Definition at line 19 of file ROperator_Softmax.hxx.