1#ifndef TMVA_SOFIE_ROPERATOR_RESHAPE
2#define TMVA_SOFIE_ROPERATOR_RESHAPE
12namespace Experimental{
39 :
fOpMode(opMode),
fNData(UTILITY::Clean_name(nameData)),
fNShape(UTILITY::Clean_name(nameShape)),
40 fNOutput(UTILITY::Clean_name(nameOutput))
57 auto ret = std::vector<ETensorType>(1,
input[0]);
63 std::vector<std::vector<size_t>> ret;
64 auto & input_shape =
input[0];
67 if (
input.size() != 2)
throw std::runtime_error(
"TMVA SOFIE Reshape Op needs 2 input tensors");
68 auto output_shape =
input[1];
72 if (input_length != output_length) {
73 if (output_shape.size() > 1 && ((output_length == 0 &&
fAllowZero == 0) || output_length >
INT64_MAX)) {
75 for (
size_t i = 0; i < output_shape.size(); i++) {
76 if (output_shape[i] == 0 || output_shape[i] ==
static_cast<size_t>(-1)) {
77 auto tmp = output_shape;
78 tmp.erase(tmp.begin() + i);
80 output_shape[i] = input_length / tmp_length;
86 throw std::runtime_error(
"TMVA Reshape Op : Invalid shapes : " +
ConvertShapeToString(input_shape) +
90 ret.push_back(output_shape);
96 std::vector<size_t> newShape = {
b, inputSize /
b};
97 ret.push_back(newShape);
102 auto output_shape =
input[0];
103 if (
input.size() == 1) {
105 while (i < output_shape.size()) {
106 if (output_shape[i] == 1 ) {
107 output_shape.erase(output_shape.begin() + i);
113 }
else if (
input.size() == 2) {
114 auto & axes =
input[1];
115 for (
size_t i = 0; i < axes.size(); i++){
116 if (output_shape[axes[i]] != 1)
119 output_shape.erase(output_shape.begin() + axes[i]);
122 ret.push_back(output_shape);
127 assert(
input.size() == 2);
128 auto output_shape =
input[0];
129 auto &axes =
input[1];
131 for (
auto & i : axes)
132 output_shape.insert(output_shape.begin() + i, 1);
135 for (
auto &i : axes) {
137 output_shape.insert(output_shape.begin() + (output_shape.size() + i - 1), 1);
140 ret.push_back(output_shape);
150 throw std::runtime_error(
"TMVA Reshape Op Input Tensor " +
fNData +
" is not found in model");
158 auto input_shape =
static_cast<int64_t *
>(dptr.get());
160 assert(
vec.size() == 1);
163 std::vector<size_t> descShape(
n);
164 std::copy(input_shape, input_shape +
n, descShape.begin());
167 throw std::runtime_error(
"TMVA Reshape Op Shape Tensor " +
fNShape +
" is not found in model");
171 std::vector<size_t> descShape(
fAttrAxes.size());
177 throw std::runtime_error(
"TMVA Reshape Op : Invalid Input/Attribute data");
184 OpName =
"op_" + OpName;
186 throw std::runtime_error(
"TMVA SOFIE Reshape Op called to Generate without being initialized first");
192 throw std::runtime_error(
"TMVA SOFIE Reshape Op : wrong output shape - is " +
199 std::stringstream out;
200 std::string opName =
"Reshape";
206 opName =
"Unsquueze";
208 out <<
SP <<
"///--------" << opName <<
" operator\n" << std::endl;
209 out <<
SP <<
"std::copy( fTensor_" <<
fNData <<
".begin(), fTensor_" <<
fNData <<
".end(), fTensor_" <<
fNOutput
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)
void AddIntermediateTensor(std::string tensor_name, ETensorType type, std::vector< std::size_t > shape)
bool CheckIfTensorAlreadyExist(std::string tensor_name)
const std::vector< size_t > & GetTensorShape(std::string name)
std::shared_ptr< void > GetInitializedTensorData(std::string tensor_name)
ROperator_Reshape(ReshapeOpMode opMode, std::vector< int64_t > attrAxes, std::string nameData, std::string nameOutput)
std::vector< int64_t > fAttrAxes
std::vector< size_t > fShapeOutput
ROperator_Reshape(ReshapeOpMode opMode, int attr_value, std::string nameData, std::string nameShape, std::string nameOutput)
std::vector< std::vector< size_t > > ShapeInference(std::vector< std::vector< size_t > > input)
std::vector< ETensorType > TypeInference(std::vector< ETensorType > input)
std::vector< size_t > fShapeInput
void Initialize(RModel &model)
std::string Generate(std::string OpName)
const std::string SP
space used to correctly indent the generated C++ code
std::string ConvertShapeToString(std::vector< size_t > shape)
std::size_t ConvertShapeToLength(std::vector< size_t > shape)
create variable transformations