Logo ROOT  

Reference Guide
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::ROperator_Conv< T > Class Template Referencefinal

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

Definition at line 20 of file ROperator_Conv.hxx.

Public Member Functions

 ROperator_Conv ()
 ROperator_Conv (std::string autopad, std::vector< size_t > dilations, size_t group, std::vector< size_t > kernelShape, std::vector< size_t > pads, std::vector< size_t > strides, std::string nameX, std::string nameW, std::string nameB, std::string nameY)
 ROperator_Conv (std::string autopad, std::vector< size_t > dilations, size_t group, std::vector< size_t > kernelShape, std::vector< size_t > pads, std::vector< size_t > strides, std::string nameX, std::string nameW, std::string nameY)
std::vector< Dim > DoShapeInference (const std::vector< Dim > &input, const std::vector< size_t > &weight)
std::string Generate (std::string OpName) override
virtual std::string GenerateDeclCode ()
std::string GenerateInitCode () override
virtual std::string GenerateSessionMembersCode (std::string)
std::vector< std::string > GetBlasRoutines () override
 Returns the blas routines needed to compile the generated code.
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
virtual std::vector< std::vector< size_t > > ShapeInference (std::vector< std::vector< size_t > >)
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

std::string convK
std::string fAttrAutopad
std::vector< size_t > fAttrDilations
size_t fAttrGroup
std::vector< size_t > fAttrKernelShape
std::vector< size_t > fAttrPads
std::vector< size_t > fAttrStrides
bool fBroadcastBias = false
size_t fDim
std::string fNB
std::string fNW
std::string fNX
std::string fNY
std::vector< size_t > fShapeB
std::vector< size_t > fShapeW
std::vector< Dim > fShapeX
std::vector< Dim > fShapeY
std::string fType
std::string imcol

#include <TMVA/ROperator_Conv.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_Conv< T >:
TMVA::Experimental::SOFIE::ROperator

Constructor & Destructor Documentation

◆ ROperator_Conv() [1/3]

template<typename T>
TMVA::Experimental::SOFIE::ROperator_Conv< T >::ROperator_Conv ( )
inline

Definition at line 52 of file ROperator_Conv.hxx.

◆ ROperator_Conv() [2/3]

template<typename T>
TMVA::Experimental::SOFIE::ROperator_Conv< T >::ROperator_Conv ( std::string autopad,
std::vector< size_t > dilations,
size_t group,
std::vector< size_t > kernelShape,
std::vector< size_t > pads,
std::vector< size_t > strides,
std::string nameX,
std::string nameW,
std::string nameB,
std::string nameY )
inline

Definition at line 54 of file ROperator_Conv.hxx.

◆ ROperator_Conv() [3/3]

template<typename T>
TMVA::Experimental::SOFIE::ROperator_Conv< T >::ROperator_Conv ( std::string autopad,
std::vector< size_t > dilations,
size_t group,
std::vector< size_t > kernelShape,
std::vector< size_t > pads,
std::vector< size_t > strides,
std::string nameX,
std::string nameW,
std::string nameY )
inline

Definition at line 73 of file ROperator_Conv.hxx.

Member Function Documentation

◆ DoShapeInference()

template<typename T>
std::vector< Dim > TMVA::Experimental::SOFIE::ROperator_Conv< T >::DoShapeInference ( const std::vector< Dim > & input,
const std::vector< size_t > & weight )
inline

Definition at line 97 of file ROperator_Conv.hxx.

◆ Generate()

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::Generate ( std::string OpName)
inlineoverridevirtual

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

Definition at line 363 of file ROperator_Conv.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()

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::GenerateInitCode ( )
inlineoverridevirtual

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

Definition at line 335 of file ROperator_Conv.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()

template<typename T>
std::vector< std::string > TMVA::Experimental::SOFIE::ROperator_Conv< T >::GetBlasRoutines ( )
inlineoverridevirtual

Returns the blas routines needed to compile the generated code.

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

Definition at line 620 of file ROperator_Conv.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()

template<typename T>
void TMVA::Experimental::SOFIE::ROperator_Conv< T >::Initialize ( RModel & model)
inlineoverridevirtual

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

Definition at line 238 of file ROperator_Conv.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.

◆ ShapeInference()

virtual std::vector< std::vector< size_t > > TMVA::Experimental::SOFIE::ROperator::ShapeInference ( std::vector< std::vector< size_t > > )
inlinevirtualinherited

Reimplemented in 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_BatchNormalization< T >, TMVA::Experimental::SOFIE::ROperator_Cast, TMVA::Experimental::SOFIE::ROperator_Comparision< T, Op >, TMVA::Experimental::SOFIE::ROperator_Concat, TMVA::Experimental::SOFIE::ROperator_Constant< T >, TMVA::Experimental::SOFIE::ROperator_ConvTranspose< T >, TMVA::Experimental::SOFIE::ROperator_Custom< T >, TMVA::Experimental::SOFIE::ROperator_Einsum< T >, TMVA::Experimental::SOFIE::ROperator_Elu< T >, TMVA::Experimental::SOFIE::ROperator_Erf< T >, TMVA::Experimental::SOFIE::ROperator_EyeLike, TMVA::Experimental::SOFIE::ROperator_Gather, TMVA::Experimental::SOFIE::ROperator_Gemm< T >, TMVA::Experimental::SOFIE::ROperator_GRU< T >, TMVA::Experimental::SOFIE::ROperator_Identity< T >, TMVA::Experimental::SOFIE::ROperator_If, TMVA::Experimental::SOFIE::ROperator_LayerNormalization< T >, TMVA::Experimental::SOFIE::ROperator_LeakyRelu< T >, TMVA::Experimental::SOFIE::ROperator_LSTM< T >, TMVA::Experimental::SOFIE::ROperator_Pad< T >, TMVA::Experimental::SOFIE::ROperator_Pool< T >, TMVA::Experimental::SOFIE::ROperator_Random, TMVA::Experimental::SOFIE::ROperator_Relu< T >, TMVA::Experimental::SOFIE::ROperator_Reshape, TMVA::Experimental::SOFIE::ROperator_RNN< T >, TMVA::Experimental::SOFIE::ROperator_Selu< T >, TMVA::Experimental::SOFIE::ROperator_Sigmoid< T >, TMVA::Experimental::SOFIE::ROperator_Softmax, TMVA::Experimental::SOFIE::ROperator_Split, TMVA::Experimental::SOFIE::ROperator_Swish< T >, TMVA::Experimental::SOFIE::ROperator_Tanh< T >, TMVA::Experimental::SOFIE::ROperator_Transpose, and TMVA::Experimental::SOFIE::ROperator_Where< T >.

Definition at line 24 of file ROperator.hxx.

◆ TypeInference()

template<typename T>
std::vector< ETensorType > TMVA::Experimental::SOFIE::ROperator_Conv< T >::TypeInference ( std::vector< ETensorType > input)
inlineoverridevirtual

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

Definition at line 91 of file ROperator_Conv.hxx.

Member Data Documentation

◆ convK

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::convK
private

Definition at line 37 of file ROperator_Conv.hxx.

◆ fAttrAutopad

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrAutopad
private

Definition at line 25 of file ROperator_Conv.hxx.

◆ fAttrDilations

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrDilations
private

Definition at line 26 of file ROperator_Conv.hxx.

◆ fAttrGroup

template<typename T>
size_t TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrGroup
private

Definition at line 27 of file ROperator_Conv.hxx.

◆ fAttrKernelShape

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrKernelShape
private

Definition at line 28 of file ROperator_Conv.hxx.

◆ fAttrPads

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrPads
private

Definition at line 29 of file ROperator_Conv.hxx.

◆ fAttrStrides

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fAttrStrides
private

Definition at line 30 of file ROperator_Conv.hxx.

◆ fBroadcastBias

template<typename T>
bool TMVA::Experimental::SOFIE::ROperator_Conv< T >::fBroadcastBias = false
private

Definition at line 23 of file ROperator_Conv.hxx.

◆ fDim

template<typename T>
size_t TMVA::Experimental::SOFIE::ROperator_Conv< T >::fDim
private

Definition at line 47 of file ROperator_Conv.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.

◆ fNB

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fNB
private

Definition at line 34 of file ROperator_Conv.hxx.

◆ fNW

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fNW
private

Definition at line 33 of file ROperator_Conv.hxx.

◆ fNX

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fNX
private

Definition at line 32 of file ROperator_Conv.hxx.

◆ fNY

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fNY
private

Definition at line 35 of file ROperator_Conv.hxx.

◆ fOutputTensorNames

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

Definition at line 51 of file ROperator.hxx.

◆ fShapeB

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fShapeB
private

Definition at line 42 of file ROperator_Conv.hxx.

◆ fShapeW

template<typename T>
std::vector<size_t> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fShapeW
private

Definition at line 41 of file ROperator_Conv.hxx.

◆ fShapeX

template<typename T>
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fShapeX
private

Definition at line 40 of file ROperator_Conv.hxx.

◆ fShapeY

template<typename T>
std::vector<Dim> TMVA::Experimental::SOFIE::ROperator_Conv< T >::fShapeY
private

Definition at line 43 of file ROperator_Conv.hxx.

◆ fType

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::fType
private

Definition at line 45 of file ROperator_Conv.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.

◆ imcol

template<typename T>
std::string TMVA::Experimental::SOFIE::ROperator_Conv< T >::imcol
private

Definition at line 38 of file ROperator_Conv.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: