@@ -44,8 +44,8 @@ public:
4444 }
4545 fShapeX = model.GetDimTensorShape (fNX );
4646 if (model.Verbose ())
47- std::cout << " GatherND - initial shape " << ConvertShapeToString (fShapeX ) << " shape of indices "
48- << ConvertShapeToString (model.GetDimTensorShape (fNIndices )) << std::endl;
47+ std::cout << " GatherND - initial shape " << ConvertDimShapeToString (fShapeX ) << " shape of indices "
48+ << ConvertDimShapeToString (model.GetDimTensorShape (fNIndices )) << std::endl;
4949 // fShapeIndices can be dynamic
5050 fShapeIndices = model.GetDimTensorShape (fNIndices );
5151 size_t q = fShapeIndices .size ();
@@ -64,8 +64,8 @@ public:
6464 if (fBatchDims > 0 ) {
6565 for (size_t i = 0 ; i < fBatchDims ; i++) {
6666 if (fShapeX [i] != fShapeIndices [i]) {
67- std::cout << " input shape " << ConvertShapeToString (fShapeX ) << " "
68- << " index shape " << ConvertShapeToString (fShapeIndices ) << std::endl;
67+ std::cout << " input shape " << ConvertDimShapeToString (fShapeX ) << " "
68+ << " index shape " << ConvertDimShapeToString (fShapeIndices ) << std::endl;
6969 throw std::runtime_error (" TMVA SOFIE GatherND : invalid input or index shape for " + std::to_string (i));
7070 }
7171 }
@@ -89,9 +89,9 @@ public:
8989 fShapeY = std::vector<Dim>(fShapeIndices .begin (), fShapeIndices .end () - 1 );
9090 fShapeY .insert (fShapeY .end (), fShapeX .begin () + fBatchDims + last_index_shape, fShapeX .end ());
9191 if (fShapeY .size () != output_rank) {
92- std::cout << " input shape " << ConvertShapeToString (fShapeX ) << " "
93- << " index shape " << ConvertShapeToString (fShapeIndices )
94- << " output shape " << ConvertShapeToString (fShapeY )
92+ std::cout << " input shape " << ConvertDimShapeToString (fShapeX ) << " "
93+ << " index shape " << ConvertDimShapeToString (fShapeIndices )
94+ << " output shape " << ConvertDimShapeToString (fShapeY )
9595 << " and output rank should be " << output_rank << std::endl;
9696 throw std::runtime_error (" TMVA SOFIE GatherND : Something is wrong in initialization " );
9797 }
@@ -101,8 +101,8 @@ public:
101101 model.AddIntermediateTensor (fNY , model.GetTensorType (fNX ), fShapeY );
102102 fType = ConvertTypeToString (model.GetTensorType (fNX ));
103103 if (model.Verbose ())
104- std::cout << " GatherND: input " << fNX << " " << ConvertShapeToString (fShapeX ) << " indices " << fNIndices << ConvertShapeToString (fShapeIndices )
105- << " -> " << fNY << " with shape " << ConvertShapeToString (fShapeY ) << std::endl;
104+ std::cout << " GatherND: input " << fNX << " " << ConvertDimShapeToString (fShapeX ) << " indices " << fNIndices << ConvertDimShapeToString (fShapeIndices )
105+ << " -> " << fNY << " with shape " << ConvertDimShapeToString (fShapeY ) << std::endl;
106106 }
107107
108108
@@ -176,7 +176,7 @@ public:
176176 }
177177 opName = " op_" + opName;
178178 std::stringstream out;
179- out << " //--------- GatherND " << opName << " --> " << ConvertShapeToString (fShapeY ) << " \n " ;
179+ out << " //--------- GatherND " << opName << " --> " << ConvertDimShapeToString (fShapeY ) << " \n " ;
180180 // The shape of the output is q + r - 1
181181 size_t r = fShapeX .size ();
182182 // Indices of shape q
0 commit comments