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) override

void	Initialize (RModel &model) override

std::string	Name ()

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
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
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_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 )

inlineoverridevirtual

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

Definition at line 123 of file ROperator_Reduce.hxx.

◆ Initialize()

template<typename T , EReduceOpMode Op>

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

inlineoverridevirtual

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

Definition at line 92 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 )