Definition at line 15 of file ROperator_Pad.hxx.
Public Types | |
enum | EMode { kConstant , kReflect , kEdge , kWrap } |
Public Member Functions | |
ROperator_Pad () | |
ROperator_Pad (const std::string &nameX, const std::string &nameP, const std::string &nameCV, const std::string &nameAX, const std::string &nameY, const std::string &mode) | |
std::string | Generate (std::string OpName) 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 |
![]() | |
virtual | ~ROperator () |
virtual std::string | GenerateDeclCode () |
virtual std::string | GenerateInitCode () |
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 Attributes | |
T | fConstantValue |
std::vector< size_t > | fInputShape |
EMode | fMode |
std::string | fNAX |
std::string | fNCV |
std::string | fNP |
std::string | fNX |
std::string | fNY |
std::vector< size_t > | fOutputShape |
std::vector< std::pair< int64_t, int64_t > > | fPads |
Additional Inherited Members | |
![]() | |
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_Pad.hxx>
enum TMVA::Experimental::SOFIE::ROperator_Pad::EMode |
Enumerator | |
---|---|
kConstant | |
kReflect | |
kEdge | |
kWrap |
Definition at line 18 of file ROperator_Pad.hxx.
|
inline |
Definition at line 34 of file ROperator_Pad.hxx.
|
inline |
Definition at line 35 of file ROperator_Pad.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 151 of file ROperator_Pad.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 64 of file ROperator_Pad.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 59 of file ROperator_Pad.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 55 of file ROperator_Pad.hxx.
|
private |
Definition at line 26 of file ROperator_Pad.hxx.
|
private |
Definition at line 28 of file ROperator_Pad.hxx.
|
private |
Definition at line 27 of file ROperator_Pad.hxx.
|
private |
Definition at line 24 of file ROperator_Pad.hxx.
|
private |
Definition at line 23 of file ROperator_Pad.hxx.
|
private |
Definition at line 22 of file ROperator_Pad.hxx.
|
private |
Definition at line 21 of file ROperator_Pad.hxx.
|
private |
Definition at line 25 of file ROperator_Pad.hxx.
|
private |
Definition at line 29 of file ROperator_Pad.hxx.
|
private |
Definition at line 30 of file ROperator_Pad.hxx.