Definition at line 15 of file ROperator_ScatterElements.hxx.

Public Member Functions
	ROperator_ScatterElements ()

	ROperator_ScatterElements (const std::string &nameX, const std::string &nameI, const std::string &nameU, const std::string &nameY, int axis, std::string reduction)

std::string	Generate (std::string opName) override

std::string	GenerateInitCode () override

void	Initialize (RModel &model) override

std::vector< std::vector< size_t > >	ShapeInference (std::vector< std::vector< size_t > > input) override

std::vector< ETensorType >	TypeInference (std::vector< ETensorType > input) override

Public Member Functions inherited from TMVA::Experimental::SOFIE::ROperator
virtual	~ROperator ()

virtual std::string	GenerateDeclCode ()

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 ()

Private Member Functions
std::string	ReductionFunction (const std::string &t1, const std::string &t2)

Private Attributes
int64_t	fAxis

std::string	fNI

std::string	fNU

std::string	fNX

std::string	fNY

std::string	fReduction

std::vector< size_t >	fShapeI

std::vector< size_t >	fShapeX

std::vector< size_t >	fShapeY

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)

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

Inheritance diagram for TMVA::Experimental::SOFIE::ROperator_ScatterElements:

[legend]

Constructor & Destructor Documentation

◆ ROperator_ScatterElements() [1/2]

TMVA::Experimental::SOFIE::ROperator_ScatterElements::ROperator_ScatterElements ( )

inline

Definition at line 51 of file ROperator_ScatterElements.hxx.

◆ ROperator_ScatterElements() [2/2]

TMVA::Experimental::SOFIE::ROperator_ScatterElements::ROperator_ScatterElements	(	const std::string &	nameX,
		const std::string &	nameI,
		const std::string &	nameU,
		const std::string &	nameY,
		int	axis,
		std::string	reduction )

inline

Definition at line 52 of file ROperator_ScatterElements.hxx.

Member Function Documentation

◆ Generate()

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

inlineoverridevirtual

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

Definition at line 108 of file ROperator_ScatterElements.hxx.

◆ GenerateInitCode()

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::GenerateInitCode ( )

inlineoverridevirtual

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

Definition at line 103 of file ROperator_ScatterElements.hxx.

◆ Initialize()

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

inlineoverridevirtual

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

Definition at line 74 of file ROperator_ScatterElements.hxx.

◆ ReductionFunction()

std::string TMVA::Experimental::SOFIE::ROperator_ScatterElements::ReductionFunction	(	const std::string &	t1,
		const std::string &	t2 )

inlineprivate

Definition at line 32 of file ROperator_ScatterElements.hxx.

◆ ShapeInference()

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

inlineoverridevirtual

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

Definition at line 69 of file ROperator_ScatterElements.hxx.