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::string | Generate (std::string OpName) |
std::string | GenerateInitCode () |
virtual std::string | GenerateSessionMembersCode (std::string opName) |
std::vector< std::string > | GetBlasRoutines () |
Returns the blas routines needed to compile the generated code. | |
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::vector< std::string > | GetStdLibs () |
virtual std::string | Header () |
Private Attributes | |
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 |
size_t | fDim |
std::string | fNB |
std::string | fNB2 |
std::string | fNW |
std::string | fNX |
std::string | fNY |
std::vector< size_t > | fShapeB |
std::vector< size_t > | fShapeW |
std::vector< size_t > | fShapeX |
std::vector< size_t > | fShapeY |
std::string | fType |
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_Conv.hxx>
|
inline |
Definition at line 48 of file ROperator_Conv.hxx.
|
inline |
Definition at line 50 of file ROperator_Conv.hxx.
|
inline |
Definition at line 67 of file ROperator_Conv.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 303 of file ROperator_Conv.hxx.
|
inlinevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 262 of file ROperator_Conv.hxx.
|
inlinevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 281 of file ROperator_Conv.hxx.
|
inlinevirtual |
Returns the blas routines needed to compile the generated code.
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 541 of file ROperator_Conv.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 197 of file ROperator_Conv.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 89 of file ROperator_Conv.hxx.
|
inlinevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 83 of file ROperator_Conv.hxx.
|
private |
Definition at line 23 of file ROperator_Conv.hxx.
|
private |
Definition at line 24 of file ROperator_Conv.hxx.
|
private |
Definition at line 25 of file ROperator_Conv.hxx.
|
private |
Definition at line 26 of file ROperator_Conv.hxx.
|
private |
Definition at line 27 of file ROperator_Conv.hxx.
|
private |
Definition at line 28 of file ROperator_Conv.hxx.
|
private |
Definition at line 43 of file ROperator_Conv.hxx.
|
private |
Definition at line 32 of file ROperator_Conv.hxx.
|
private |
Definition at line 33 of file ROperator_Conv.hxx.
|
private |
Definition at line 31 of file ROperator_Conv.hxx.
|
private |
Definition at line 30 of file ROperator_Conv.hxx.
|
private |
Definition at line 34 of file ROperator_Conv.hxx.
|
private |
Definition at line 38 of file ROperator_Conv.hxx.
|
private |
Definition at line 37 of file ROperator_Conv.hxx.
|
private |
Definition at line 36 of file ROperator_Conv.hxx.
|
private |
Definition at line 39 of file ROperator_Conv.hxx.
|
private |
Definition at line 41 of file ROperator_Conv.hxx.