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) override |
| std::string | GenerateInitCode () override |
| std::vector< std::string > | GetBlasRoutines () override |
| Returns the blas routines needed to compile the generated code. | |
| 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) |
| 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 | |
| std::string | convK |
| 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 |
| std::string | imcol |
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_Conv.hxx>
|
inline |
Definition at line 51 of file ROperator_Conv.hxx.
|
inline |
Definition at line 53 of file ROperator_Conv.hxx.
|
inline |
Definition at line 72 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 303 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 285 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Returns the blas routines needed to compile the generated code.
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 527 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Implements TMVA::Experimental::SOFIE::ROperator.
Definition at line 204 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 96 of file ROperator_Conv.hxx.
|
inlineoverridevirtual |
Reimplemented from TMVA::Experimental::SOFIE::ROperator.
Definition at line 90 of file ROperator_Conv.hxx.
|
private |
Definition at line 36 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 46 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 41 of file ROperator_Conv.hxx.
|
private |
Definition at line 40 of file ROperator_Conv.hxx.
|
private |
Definition at line 39 of file ROperator_Conv.hxx.
|
private |
Definition at line 42 of file ROperator_Conv.hxx.
|
private |
Definition at line 44 of file ROperator_Conv.hxx.
|
private |
Definition at line 37 of file ROperator_Conv.hxx.