Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ROperator_BasicUnary.hxx
Go to the documentation of this file.
1#ifndef TMVA_EXPERIMENTAL_SOFIE_ROPERATOR_BASIC_UNARY
2#define TMVA_EXPERIMENTAL_SOFIE_ROPERATOR_BASIC_UNARY
3
4#include <TMVA/ROperator.hxx>
5#include <TMVA/RModel.hxx>
7
8namespace TMVA {
9namespace Experimental {
10namespace SOFIE {
11
13
14template <typename T, EBasicUnaryOperator Op>
16};
17
18template <typename T>
20 static std::string Name() { return "Reciprocal"; }
21 static std::string Op(const std::string &X) { return "1/" + X; }
22};
23
24template <typename T>
26 static std::string Name() { return "Sqrt"; }
27 static std::string Op(const std::string &X) { return "std::sqrt(" + X + ")"; }
28};
29
30template <typename T>
32 static std::string Name() { return "Neg"; }
33 static std::string Op(const std::string &X) { return "-" + X; }
34};
35
36template <typename T>
38 static std::string Name() { return "Exp"; }
39 static std::string Op(const std::string &X) { return "std::exp(" + X + ")"; }
40};
41
42template <typename T>
44 static std::string Name() { return "Log"; }
45 static std::string Op(const std::string &X) { return "std::log(" + X + ")"; }
46};
47
48template <typename T, EBasicUnaryOperator Op>
49class ROperator_BasicUnary final : public ROperator {
50private:
51 std::string fNX;
52 std::string fNY;
53
54 std::vector<size_t> fShapeX;
55 std::vector<size_t> fShapeY;
56
57public:
59
60 ROperator_BasicUnary(std::string nameX, std::string nameY)
61 : fNX(UTILITY::Clean_name(nameX)), fNY(UTILITY::Clean_name(nameY))
62 {}
63
64 std::vector<std::vector<size_t>> ShapeInference(std::vector<std::vector<size_t>> input) override { return input; }
65
66 std::vector<ETensorType> TypeInference(std::vector<ETensorType> input) override { return input; }
67
68 void Initialize(RModel &model) override
69 {
70 if (!model.CheckIfTensorAlreadyExist(fNX)) {
71 throw std::runtime_error("TMVA::SOFIE - Tensor " + fNX + " not found.");
72 }
73 fShapeX = model.GetTensorShape(fNX);
76 }
77
78 std::string Generate(std::string OpName) override
79 {
80 OpName = "op_" + OpName;
81 std::stringstream out;
82
83 out << SP << "\n//---- Operator" << UnaryOpTraits<T, Op>::Name() << " " << OpName << "\n";
85 out << SP << "for (size_t i = 0; i < " << length << "; i++) {\n";
86 out << SP << SP << "tensor_" << fNY << "[i] = " << UnaryOpTraits<T, Op>::Op("tensor_" + fNX + "[i]") << ";\n";
87 out << SP << "}\n";
88 return out.str();
89 }
90};
91
92} // namespace SOFIE
93} // namespace Experimental
94} // namespace TMVA
95
96#endif
#define X(type, name)
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void input
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h length
const ETensorType & GetTensorType(std::string name)
Definition RModel.cxx:94
void AddIntermediateTensor(std::string tensor_name, ETensorType type, std::vector< Dim > dim_shape)
Definition RModel.cxx:203
bool CheckIfTensorAlreadyExist(std::string tensor_name)
Definition RModel.cxx:122
const std::vector< size_t > & GetTensorShape(std::string name)
Definition RModel.cxx:56
std::vector< ETensorType > TypeInference(std::vector< ETensorType > input) override
std::string Generate(std::string OpName) override
ROperator_BasicUnary(std::string nameX, std::string nameY)
std::vector< std::vector< size_t > > ShapeInference(std::vector< std::vector< size_t > > input) override
const std::string SP
space used to correctly indent the generated C++ code
Definition ROperator.hxx:41
std::size_t ConvertShapeToLength(std::vector< size_t > shape)
create variable transformations