@@ -230,7 +230,8 @@ int main(int argc, char* argv[])
230230 std::cout << " Number of Input Nodes: " << numInputNodes << std::endl;
231231 std::cout << " Number of Output Nodes: " << numOutputNodes << std::endl;
232232
233- const char * inputName = session.GetInputName (0 , allocator);
233+ auto inputNodeName = session.GetInputNameAllocated (0 , allocator);
234+ const char * inputName = inputNodeName.get ();
234235 std::cout << " Input Name: " << inputName << std::endl;
235236
236237 Ort::TypeInfo inputTypeInfo = session.GetInputTypeInfo (0 );
@@ -242,7 +243,8 @@ int main(int argc, char* argv[])
242243 std::vector<int64_t > inputDims = inputTensorInfo.GetShape ();
243244 std::cout << " Input Dimensions: " << inputDims << std::endl;
244245
245- const char * outputName = session.GetOutputName (0 , allocator);
246+ auto outputNodeName = session.GetOutputNameAllocated (0 , allocator);
247+ const char * outputName = outputNodeName.get ();
246248 std::cout << " Output Name: " << outputName << std::endl;
247249
248250 Ort::TypeInfo outputTypeInfo = session.GetOutputTypeInfo (0 );
@@ -311,7 +313,7 @@ int main(int argc, char* argv[])
311313 so we have to prepare an array of Ort::Value instances for inputs and outputs respectively even if
312314 we only have one input and one output. */
313315
314- std::vector<const char *> inputNames{inputName };
316+ std::vector<const char *> inputNames{" data_0 " };
315317 std::vector<const char *> outputNames{outputName};
316318 std::vector<Ort::Value> inputTensors;
317319 std::vector<Ort::Value> outputTensors;
0 commit comments