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

Definition at line 18 of file ROperator_Reshape.hxx.

Public Member Functions
	ROperator_Reshape ()

	ROperator_Reshape (ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameShape, std::string nameOutput)

	ROperator_Reshape (ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput)

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< 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	fAllowZero = 0

std::vector< int64_t >	fAttrAxes

int	fAxis = 1

std::string	fNData

std::string	fNOutput

std::string	fNShape

ReshapeOpMode	fOpMode = Reshape

std::vector< size_t >	fShapeInput

std::vector< size_t >	fShapeOutput

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

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

[legend]

Constructor & Destructor Documentation

◆ ROperator_Reshape() [1/3]

template<typename T >

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

inline

Definition at line 37 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [2/3]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_Reshape< T >::ROperator_Reshape	(	ReshapeOpMode	opMode,
		int	attr_value,
		std::string	nameData,
		std::string	nameShape,
		std::string	nameOutput
	)

inline

Definition at line 38 of file ROperator_Reshape.hxx.

◆ ROperator_Reshape() [3/3]

template<typename T >

TMVA::Experimental::SOFIE::ROperator_Reshape< T >::ROperator_Reshape	(	ReshapeOpMode	opMode,
		std::vector< int64_t >	attrAxes,
		std::string	nameData,
		std::string	nameOutput
	)

inline

Definition at line 48 of file ROperator_Reshape.hxx.

Member Function Documentation

◆ Generate()

template<typename T >

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

inlinevirtual

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

Definition at line 181 of file ROperator_Reshape.hxx.

◆ Initialize()

template<typename T >

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

inlinevirtual

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

Definition at line 145 of file ROperator_Reshape.hxx.

◆ ShapeInference()

template<typename T >

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

inlinevirtual

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

Definition at line 62 of file ROperator_Reshape.hxx.