Skip to content

Commit b5e3934

Browse files
author
nmaajidk
committed
Samples updated
1 parent aadb8fc commit b5e3934

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

c_cxx/OpenVINO_EP/Linux/squeezenet_classification/squeezenet_cpp_app.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)