Definition at line 18 of file ROperator_Reshape.hxx.
Public Member Functions | |
ROperator_Reshape () | |
ROperator_Reshape (ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameShape, std::string nameOutput) | |
ROperator_Reshape (ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput) | |
std::string | Generate (std::string OpName) |
void | Initialize (RModel &model) |
std::vector< std::vector< size_t > > | ShapeInference (std::vector< std::vector< size_t > > input) |
std::vector< ETensorType > | TypeInference (std::vector< ETensorType > input) |
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 | |
int | fAllowZero = 0 |
std::vector< int64_t > | fAttrAxes |
int | fAxis = 1 |
std::string | fNData |
std::string | fNOutput |
std::string | fNShape |
ReshapeOpMode | fOpMode = Reshape |
std::vector< size_t > | fShapeInput |
std::vector< size_t > | fShapeOutput |
Additional Inherited Members | |
Protected Attributes inherited from TMVA::Experimental::SOFIE::ROperator | |
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>
|
inline |
Definition at line 37 of file ROperator_Reshape.hxx.
|
inline |
Definition at line 38 of file ROperator_Reshape.hxx.
|
inline |
Definition at line 48 of file ROperator_Reshape.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 182 of file ROperator_Reshape.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 145 of file ROperator_Reshape.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 62 of file ROperator_Reshape.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 56 of file ROperator_Reshape.hxx.
|
private |
Definition at line 25 of file ROperator_Reshape.hxx.
|
private |
Definition at line 33 of file ROperator_Reshape.hxx.
|
private |
Definition at line 26 of file ROperator_Reshape.hxx.
|
private |
Definition at line 28 of file ROperator_Reshape.hxx.
|
private |
Definition at line 30 of file ROperator_Reshape.hxx.
|
private |
Definition at line 29 of file ROperator_Reshape.hxx.
|
private |
Definition at line 23 of file ROperator_Reshape.hxx.
|
private |
Definition at line 31 of file ROperator_Reshape.hxx.
|
private |
Definition at line 32 of file ROperator_Reshape.hxx.