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::string	Generate (std::string opName) override
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 ()
void	Initialize (RModel &model) override
bool	IsOutputConstant () const
	check if the output of the operator is Constant and is evaluated at initialization time
std::string	Name () const
std::vector< std::vector< Dim > >	ShapeInference (const std::vector< std::vector< Dim > > &input)
std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > > input) override
std::vector< ETensorType >	TypeInference (std::vector< ETensorType > input) override

Protected Attributes
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

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< int64_t >	fShape
std::vector< Dim >	fShapeInput
std::vector< Dim >	fShapeOutput
bool	fVerbose = false

#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 51 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 52 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 68 of file ROperator_Reshape.hxx.

Member Function Documentation

Generate()

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

inlineoverridevirtual

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

Definition at line 334 of file ROperator_Reshape.hxx.

GenerateDeclCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateDeclCode ( )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_Random.

Definition at line 31 of file ROperator.hxx.

GenerateInitCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateInitCode ( )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_BasicBinary< T, Op >, TMVA::Experimental::SOFIE::ROperator_Conv< T >, TMVA::Experimental::SOFIE::ROperator_ConvTranspose< T >, TMVA::Experimental::SOFIE::ROperator_Einsum< T >, TMVA::Experimental::SOFIE::ROperator_Expand< T >, TMVA::Experimental::SOFIE::ROperator_Identity< T >, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Pool< T >, TMVA::Experimental::SOFIE::ROperator_Random, TMVA::Experimental::SOFIE::ROperator_ScatterElements, and TMVA::Experimental::SOFIE::ROperator_Where< T >.

Definition at line 29 of file ROperator.hxx.

GenerateSessionMembersCode()

virtual std::string TMVA::Experimental::SOFIE::ROperator::GenerateSessionMembersCode ( std::string )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_LSTM< T >, TMVA::Experimental::SOFIE::ROperator_NonZero< T >, TMVA::Experimental::SOFIE::ROperator_Pool< T >, and TMVA::Experimental::SOFIE::ROperator_RNN< T >.

Definition at line 33 of file ROperator.hxx.

GetBlasRoutines()

virtual std::vector< std::string > TMVA::Experimental::SOFIE::ROperator::GetBlasRoutines ( )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_BatchNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Conv< T >, TMVA::Experimental::SOFIE::ROperator_ConvTranspose< T >, TMVA::Experimental::SOFIE::ROperator_Einsum< T >, TMVA::Experimental::SOFIE::ROperator_Gemm< T >, TMVA::Experimental::SOFIE::ROperator_GRU< T >, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_LSTM< T >, and TMVA::Experimental::SOFIE::ROperator_RNN< T >.

Definition at line 22 of file ROperator.hxx.

GetOpInputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpInputTensors ( ) const

inlineinherited

Definition at line 54 of file ROperator.hxx.

GetOpOutputTensors()

std::span< const std::string_view > TMVA::Experimental::SOFIE::ROperator::GetOpOutputTensors ( ) const

inlineinherited

Definition at line 58 of file ROperator.hxx.

GetStdLibs()

virtual std::vector< std::string > TMVA::Experimental::SOFIE::ROperator::GetStdLibs ( )

inlinevirtualinherited

Reimplemented in TMVA::Experimental::SOFIE::ROperator_Basic_Is< Op >, TMVA::Experimental::SOFIE::ROperator_BasicBinary< T, Op >, TMVA::Experimental::SOFIE::ROperator_BasicNary< T, Op >, TMVA::Experimental::SOFIE::ROperator_BasicUnary< T, Op >, TMVA::Experimental::SOFIE::ROperator_Erf< T >, TMVA::Experimental::SOFIE::ROperator_Gelu, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Random, TMVA::Experimental::SOFIE::ROperator_Selu< T >, TMVA::Experimental::SOFIE::ROperator_Sigmoid< T >, TMVA::Experimental::SOFIE::ROperator_Softmax, TMVA::Experimental::SOFIE::ROperator_Swish< T >, and TMVA::Experimental::SOFIE::ROperator_Tanh< T >.

Definition at line 23 of file ROperator.hxx.

Header()

virtual std::string TMVA::Experimental::SOFIE::ROperator::Header ( )

inlinevirtualinherited

Definition at line 34 of file ROperator.hxx.

Initialize()

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

inlineoverridevirtual

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

Definition at line 246 of file ROperator_Reshape.hxx.

IsOutputConstant()

bool TMVA::Experimental::SOFIE::ROperator::IsOutputConstant ( ) const

inlineinherited

check if the output of the operator is Constant and is evaluated at initialization time

Definition at line 37 of file ROperator.hxx.

Name()

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

inline

Definition at line 43 of file ROperator_Reshape.hxx.

ShapeInference() [1/2]

std::vector< std::vector< Dim > > TMVA::Experimental::SOFIE::ROperator_Reshape::ShapeInference ( const std::vector< std::vector< Dim > > & input )

inline

Definition at line 87 of file ROperator_Reshape.hxx.

ShapeInference() [2/2]

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

inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 82 of file ROperator_Reshape.hxx.

TypeInference()

std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_Reshape::TypeInference ( std::vector< ETensorType > input )

inlineoverridevirtual

Reimplemented from TMVA::Experimental::SOFIE::ROperator.

Definition at line 78 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 38 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.

fInputTensorNames

std::vector<std::string_view> TMVA::Experimental::SOFIE::ROperator::fInputTensorNames

mutableprotectedinherited

Definition at line 50 of file ROperator.hxx.

fIsOutputConstant

bool TMVA::Experimental::SOFIE::ROperator::fIsOutputConstant = false

protectedinherited

flag to identify if operator has a constant output (no need to generate code)

Definition at line 47 of file ROperator.hxx.

fIsOutputParamShape

bool TMVA::Experimental::SOFIE::ROperator::fIsOutputParamShape = false

protectedinherited

flag to identify of the output represents a parametric shape (can be knwon at compile time)

Definition at line 48 of file ROperator.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.

fOutputTensorNames

std::vector<std::string_view> TMVA::Experimental::SOFIE::ROperator::fOutputTensorNames

mutableprotectedinherited

Definition at line 51 of file ROperator.hxx.

fShape

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

private

Definition at line 39 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.

fUseSession

bool TMVA::Experimental::SOFIE::ROperator::fUseSession = false

protectedinherited

flag to identify if using the session class

Definition at line 46 of file ROperator.hxx.

fVerbose

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

private

Definition at line 25 of file ROperator_Reshape.hxx.

SP

const std::string TMVA::Experimental::SOFIE::ROperator::SP = " "

protectedinherited

space used to correctly indent the generated C++ code

Definition at line 45 of file ROperator.hxx.

---

The documentation for this class was generated from the following file:

• tmva/sofie/inc/TMVA/ROperator_Reshape.hxx