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

Definition at line 15 of file ROperator_TopK.hxx.

Public Member Functions
	ROperator_TopK ()

	ROperator_TopK (int attr_axis, int attr_largest, int attr_sorted, std::string nameK, std::string nameX, std::string nameVal, std::string nameInd)

std::string	Generate (std::string OpName)

void	Initialize (RModel &model)

std::vector< std::vector< size_t > >	ShapeInference (const 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
int	fAttrAxis

int	fAttrLargest

int	fAttrSorted

size_t	fK

std::string	fNInd

std::string	fNK

std::string	fNVal

std::string	fNX

std::vector< size_t >	fShapeX

std::vector< size_t >	fShapeY

std::string	fType

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

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_TopK< T >:

[legend]

Constructor & Destructor Documentation

◆ ROperator_TopK() [1/2]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_TopK< T >::ROperator_TopK ( )

inline

Definition at line 32 of file ROperator_TopK.hxx.

◆ ROperator_TopK() [2/2]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_TopK< T >::ROperator_TopK	(	int	attr_axis,
		int	attr_largest,
		int	attr_sorted,
		std::string	nameK,
		std::string	nameX,
		std::string	nameVal,
		std::string	nameInd
	)

inline

Definition at line 33 of file ROperator_TopK.hxx.

Member Function Documentation

◆ Generate()

template<typename T >

std::string TMVA::Experimental::SOFIE::ROperator_TopK< T >::Generate ( std::string OpName )

inlinevirtual

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

Definition at line 109 of file ROperator_TopK.hxx.

◆ Initialize()

template<typename T >

void TMVA::Experimental::SOFIE::ROperator_TopK< T >::Initialize ( RModel & model )

inlinevirtual

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

Definition at line 60 of file ROperator_TopK.hxx.

◆ ShapeInference()

template<typename T >

std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator_TopK< T >::ShapeInference ( const std::vector< std::vector< size_t > > input )

inlinevirtual

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

Definition at line 47 of file ROperator_TopK.hxx.