TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op > Class Template Reference

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::vector< Dim >	DoShapeInference (const std::vector< Dim > &input)
std::string	Generate (std::string opName) override
void	Initialize (RModel &model) override
std::string	Name ()
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 ()
virtual std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > >)
virtual std::vector< ETensorType >	TypeInference (std::vector< ETensorType >)

Private Attributes
std::vector< int64_t >	fAttrAxes
bool	fInputDimShape = false
int	fkeepdims = 1
std::string	fNAxes
std::string	fNX
std::string	fNY
EReduceOpMode	fReduceOpMode
std::vector< Dim >	fShapeX
std::vector< Dim >	fShapeY
std::vector< Dim >	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)
bool	fIsOutputParamShape = false
	flag to identify of the output represents a parametric shape (can be knwon 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_Reduce.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >:

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 48 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 49 of file ROperator_Reduce.hxx.

Member Function Documentation

DoShapeInference()

template<typename T , EReduceOpMode Op>

std::vector< Dim > TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::DoShapeInference ( const std::vector< Dim > & input )

inline

Definition at line 62 of file ROperator_Reduce.hxx.

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 121 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 88 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 40 of file ROperator_Reduce.hxx.

Member Data Documentation

fAttrAxes

template<typename T , EReduceOpMode Op>

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fAttrAxes

private

Definition at line 28 of file ROperator_Reduce.hxx.

fInputDimShape

template<typename T , EReduceOpMode Op>

bool TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fInputDimShape = false

private

Definition at line 26 of file ROperator_Reduce.hxx.

fkeepdims

template<typename T , EReduceOpMode Op>

int TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fkeepdims = 1

private

Definition at line 27 of file ROperator_Reduce.hxx.

fNAxes

template<typename T , EReduceOpMode Op>

std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNAxes

private

Definition at line 31 of file ROperator_Reduce.hxx.

fNX

template<typename T , EReduceOpMode Op>

std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNX

private

Definition at line 30 of file ROperator_Reduce.hxx.

fNY

template<typename T , EReduceOpMode Op>

std::string TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fNY

private

Definition at line 32 of file ROperator_Reduce.hxx.

fReduceOpMode

template<typename T , EReduceOpMode Op>

EReduceOpMode TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fReduceOpMode

private

Definition at line 29 of file ROperator_Reduce.hxx.

fShapeX

template<typename T , EReduceOpMode Op>

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeX

private

Definition at line 33 of file ROperator_Reduce.hxx.

fShapeY

template<typename T , EReduceOpMode Op>

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeY

private

Definition at line 34 of file ROperator_Reduce.hxx.

fShapeYNotPruned

template<typename T , EReduceOpMode Op>

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reduce< T, Op >::fShapeYNotPruned

private

Definition at line 35 of file ROperator_Reduce.hxx.