TMVA::Experimental::SOFIE::ROperator_Reshape Class Reference

Definition at line 20 of file ROperator_Reshape.hxx.

Public Member Functions
	ROperator_Reshape ()
	ROperator_Reshape (ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameInput2, std::string nameOutput)
	ROperator_Reshape (ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput)
std::vector< Dim >	DoShapeInference (const std::vector< Dim > &input_shape, const std::vector< Dim > &target_shape)
std::string	Generate (std::string opName) override
void	Initialize (RModel &model) override
std::string	Name () const
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 ()
bool	IsOutputConstant () const
	check if the output of the operator is Constant and is evaluated at initialization time
virtual std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > >)
virtual std::vector< ETensorType >	TypeInference (std::vector< ETensorType >)

Private Attributes
int	fAllowZero = 0
std::vector< int64_t >	fAttrAxes
int	fAxis = 1
bool	fDimInput = false
bool	fDynamicShape = false
std::string	fNData
std::string	fNInput2
std::string	fNOutput
ReshapeOpMode	fOpMode = Reshape
std::vector< Dim >	fOutputShapeData
std::vector< int64_t >	fShape
std::vector< Dim >	fShapeInput
std::vector< Dim >	fShapeOutput
bool	fVerbose = 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_Reshape.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Reshape:

Constructor & Destructor Documentation

ROperator_Reshape() [1/3]

TMVA::Experimental::SOFIE::ROperator_Reshape::ROperator_Reshape ( )

inline

Definition at line 52 of file ROperator_Reshape.hxx.

ROperator_Reshape() [2/3]

TMVA::Experimental::SOFIE::ROperator_Reshape::ROperator_Reshape ( ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameInput2, std::string nameOutput )

inline

Definition at line 53 of file ROperator_Reshape.hxx.

ROperator_Reshape() [3/3]

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

inline

Definition at line 69 of file ROperator_Reshape.hxx.

Member Function Documentation

DoShapeInference()

std::vector< Dim > TMVA::Experimental::SOFIE::ROperator_Reshape::DoShapeInference ( const std::vector< Dim > & input_shape, const std::vector< Dim > & target_shape )

inline

Definition at line 80 of file ROperator_Reshape.hxx.

Generate()

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

inlineoverridevirtual

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

Definition at line 348 of file ROperator_Reshape.hxx.

Initialize()

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

inlineoverridevirtual

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

Definition at line 253 of file ROperator_Reshape.hxx.

Name()

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::Name ( ) const

inline

Definition at line 44 of file ROperator_Reshape.hxx.

Member Data Documentation

fAllowZero

int TMVA::Experimental::SOFIE::ROperator_Reshape::fAllowZero = 0

private

Definition at line 30 of file ROperator_Reshape.hxx.

fAttrAxes

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reshape::fAttrAxes

private

Definition at line 39 of file ROperator_Reshape.hxx.

fAxis

int TMVA::Experimental::SOFIE::ROperator_Reshape::fAxis = 1

private

Definition at line 31 of file ROperator_Reshape.hxx.

fDimInput

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fDimInput = false

private

Definition at line 26 of file ROperator_Reshape.hxx.

fDynamicShape

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fDynamicShape = false

private

Definition at line 27 of file ROperator_Reshape.hxx.

fNData

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNData

private

Definition at line 33 of file ROperator_Reshape.hxx.

fNInput2

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNInput2

private

Definition at line 34 of file ROperator_Reshape.hxx.

fNOutput

std::string TMVA::Experimental::SOFIE::ROperator_Reshape::fNOutput

private

Definition at line 35 of file ROperator_Reshape.hxx.

fOpMode

ReshapeOpMode TMVA::Experimental::SOFIE::ROperator_Reshape::fOpMode = Reshape

private

Definition at line 28 of file ROperator_Reshape.hxx.

fOutputShapeData

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reshape::fOutputShapeData

private

Definition at line 38 of file ROperator_Reshape.hxx.

fShape

std::vector<int64_t> TMVA::Experimental::SOFIE::ROperator_Reshape::fShape

private

Definition at line 40 of file ROperator_Reshape.hxx.

fShapeInput

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reshape::fShapeInput

private

Definition at line 36 of file ROperator_Reshape.hxx.

fShapeOutput

std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Reshape::fShapeOutput

private

Definition at line 37 of file ROperator_Reshape.hxx.

fVerbose

bool TMVA::Experimental::SOFIE::ROperator_Reshape::fVerbose = false

private

Definition at line 25 of file ROperator_Reshape.hxx.