Definition at line 15 of file ROperator_Custom.hxx.
Public Member Functions | |
ROperator_Custom () | |
ROperator_Custom (std::string OpName, std::vector< std::string >Inputs, std::vector< std::string >Outputs, std::vector< std::vector< std::size_t > > OutputShapes, std::string HeaderName) | |
std::string | Generate (std::string OpName) |
void | Initialize (RModel &model) |
std::vector< std::vector< size_t > > | ShapeInference (std::vector< std::vector< size_t > >) |
std::vector< ETensorType > | TypeInference (std::vector< ETensorType >) |
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 | |
std::string | fHeaderName |
std::vector< std::string > | fInputNames |
std::string | fOpName |
std::vector< std::string > | fOutputNames |
std::vector< std::vector< std::size_t > > | fOutputShapes |
Additional Inherited Members | |
Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator | |
bool | fIsOutputConstant = false |
flag to identify if operator has a constant output (no need to generate code) | |
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_Custom.hxx>
|
inline |
Definition at line 26 of file ROperator_Custom.hxx.
|
inline |
Definition at line 27 of file ROperator_Custom.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 67 of file ROperator_Custom.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 42 of file ROperator_Custom.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 39 of file ROperator_Custom.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 40 of file ROperator_Custom.hxx.
|
private |
Definition at line 23 of file ROperator_Custom.hxx.
|
private |
Definition at line 20 of file ROperator_Custom.hxx.
|
private |
Definition at line 19 of file ROperator_Custom.hxx.
|
private |
Definition at line 21 of file ROperator_Custom.hxx.
|
private |
Definition at line 22 of file ROperator_Custom.hxx.