1#ifndef TMVA_SOFIE_ROperator_Where
2#define TMVA_SOFIE_ROperator_Where
11namespace Experimental{
54 auto ret = std::vector<std::vector<size_t>>(1,
input[0]);
61 throw std::runtime_error(std::string(
"TMVA SOFIE Where Op Input Tensor ") +
fNA +
"is not found in model");
64 throw std::runtime_error(std::string(
"TMVA SOFIE Where Op Input Tensor ") +
fNB +
"is not found in model");
67 throw std::runtime_error(std::string(
"TMVA SOFIE Where Op Input Tensor ") +
fNC +
"is not found in model");
109 std::default_delete<T[]>());
125 std::default_delete<T[]>());
141 std::default_delete<T[]>());
162 for (
size_t i = 0; i <
dataY.size(); i++)
184 std::stringstream out;
195 throw std::runtime_error(
"TMVA SOFIE Where Op called to Generate without being initialized first");
197 std::stringstream out;
198 out <<
SP <<
"\n//-------- Where \n";
203 out <<
SP <<
"// Broadcasting uninitialized tensor " <<
fNA <<
"\n";
210 out <<
SP <<
"// Broadcasting uninitialized tensor " <<
fNB <<
"\n";
220 out <<
SP <<
"std::vector<bool> fTensor_" <<
fNC <<
"(tensor_" <<
fNC <<
", tensor_" <<
fNC <<
" + " <<
inputLength <<
");\n";
222 out <<
SP <<
"// Broadcasting uninitialized tensor " <<
fNC <<
"\n";
231 out <<
SP <<
"for (size_t id = 0; id < " <<
length <<
" ; id++){\n";
233 out <<
SP <<
SP <<
"tensor_" <<
fNY <<
"[id] = " <<
"(fTensor_" <<
nameC <<
"[id]) ? tensor_"
234 <<
nameA <<
"[id] : tensor_" +
nameB +
"[id];\n";
ROOT::Detail::TRangeCast< T, true > TRangeDynCast
TRangeDynCast is an adapter class that allows the typed iteration through a TCollection.
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void data
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< Dim > dim_shape)
bool CheckIfTensorAlreadyExist(std::string tensor_name)
void AddConstantTensor(std::string tensor_name, ETensorType type, std::vector< std::size_t > shape, std::shared_ptr< void > data)
const ETensorType & GetTensorType(std::string name) const
bool IsInitializedTensor(const std::string &name) const
const std::vector< size_t > & GetTensorShape(std::string name) const
std::shared_ptr< void > GetInitializedTensorData(std::string tensor_name)
void SetNotWritableInitializedTensor(const std::string &tensor_name)
bool IsReadyInputTensor(const std::string &name) const
std::vector< size_t > fShapeY
std::vector< size_t > fShapeC
std::vector< std::vector< size_t > > ShapeInference(std::vector< std::vector< size_t > > input) override
std::vector< ETensorType > TypeInference(std::vector< ETensorType > input) override
std::vector< size_t > fShapeB
void Initialize(RModel &model) override
std::string fNBroadcastedB
std::string Generate(std::string OpName) override
ROperator_Where(const std::string &nameA, const std::string &nameB, const std::string &nameC, const std::string &nameY)
std::string fNBroadcastedC
std::vector< size_t > fShapeA
std::string fNBroadcastedA
std::string GenerateInitCode() override
std::vector< std::string_view > fInputTensorNames
bool fIsOutputConstant
flag to identify if operator has a constant output (no need to generate code)
const std::string SP
space used to correctly indent the generated C++ code
std::vector< std::string_view > fOutputTensorNames
bool AreSameShape(const std::vector< size_t > &, const std::vector< size_t > &)
std::string ConvertValuesToString(size_t n, const T *data)
std::string ConvertShapeToString(std::vector< size_t > shape)
std::size_t ConvertShapeToLength(std::vector< size_t > shape)
create variable transformations