Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ROperator_Cast.hxx
Go to the documentation of this file.
1#ifndef TMVA_SOFIE_ROPERATOR_Cast
2#define TMVA_SOFIE_ROPERATOR_Cast
3
5#include "TMVA/ROperator.hxx"
6#include "TMVA/RModel.hxx"
7
8#include <sstream>
9
10namespace TMVA{
11namespace Experimental{
12namespace SOFIE{
13
14
15template <typename T>
16class ROperator_Cast final : public ROperator
17{
18
19private:
20
21 std::string fNX;
22 std::string fNY;
23 std::vector<size_t> fShape;
24 std::string fAttrType = "float";
25
26public:
28 ROperator_Cast(std::string attr_type,std::string nameX, std::string nameY):
29 fNX(UTILITY::Clean_name(nameX)), fNY(UTILITY::Clean_name(nameY)),
30 fAttrType(attr_type) {}
31
32 std::vector<ETensorType> TypeInference(std::vector<ETensorType> input){
33 return input;
34 }
35
36 std::vector<std::vector<size_t>> ShapeInference(std::vector<std::vector<size_t>> input){
37 auto ret = input; //suggest copy to compiler
38 return ret;
39 }
40
41 void Initialize(RModel& model){
42 //input must be a graph input, or already initialized intermediate tensor
43 if (model.CheckIfTensorAlreadyExist(fNX) == false){
44 throw std::runtime_error("TMVA SOFIE Cast Op Input Tensor is not found in model");
45 }
46 fShape = model.GetTensorShape(fNX);
48 }
49
50
51 std::string Generate(std::string OpName){
52 OpName = "op_" + OpName;
53 if (fShape.empty()) {
54 throw std::runtime_error("TMVA SOFIE Cast called to Generate without being initialized first");
55 }
56 std::stringstream out;
58
59 // out << SP << ETensorType << " " << OpName << "_attr = " << fattr << ";\n";
60 out << "\n//------ CAST\n";
61 out << SP << "for (int id = 0; id < " << length << " ; id++){\n";
62
63 out << SP << SP << "tensor_" << fNY << "[id] = static_cast<"<< fAttrType << ">(tensor_" << fNX << "[id]);\n";
64
65 out << SP << "}\n";
66 return out.str();
67 }
68
69};
70
71}//SOFIE
72}//Experimental
73}//TMVA
74
75
76#endif //TMVA_SOFIE_ROPERATOR_Cast
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
void AddIntermediateTensor(std::string tensor_name, ETensorType type, std::vector< std::size_t > shape)
Definition RModel.cxx:160
bool CheckIfTensorAlreadyExist(std::string tensor_name)
Definition RModel.cxx:101
const std::vector< size_t > & GetTensorShape(std::string name)
Definition RModel.cxx:59
std::string Generate(std::string OpName)
std::vector< ETensorType > TypeInference(std::vector< ETensorType > input)
ROperator_Cast(std::string attr_type, std::string nameX, std::string nameY)
std::vector< std::vector< size_t > > ShapeInference(std::vector< std::vector< size_t > > input)
const std::string SP
space used to correctly indent the generated C++ code
Definition ROperator.hxx:41
ETensorType ConvertStringToType(std::string type)
std::size_t ConvertShapeToLength(std::vector< size_t > shape)
create variable transformations