Skip to content

Commit 86666ce

Browse files
committed
cpp sample update
1 parent b5e3934 commit 86666ce

File tree

3 files changed

+9
-6
lines changed

3 files changed

+9
-6
lines changed

c_cxx/OpenVINO_EP/Linux/squeezenet_classification/squeezenet_cpp_app.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
/*
22
Copyright (C) 2021, Intel Corporation
33
SPDX-License-Identifier: Apache-2.0
4-
54
Portions of this software are copyright of their respective authors and released under the MIT license:
65
- ONNX-Runtime-Inference, Copyright 2020 Lei Mao. For licensing see https://github.com/leimao/ONNX-Runtime-Inference/blob/main/LICENSE.md
76
*/
@@ -313,7 +312,7 @@ int main(int argc, char* argv[])
313312
so we have to prepare an array of Ort::Value instances for inputs and outputs respectively even if
314313
we only have one input and one output. */
315314

316-
std::vector<const char*> inputNames{"data_0"};
315+
std::vector<const char*> inputNames{inputName};
317316
std::vector<const char*> outputNames{outputName};
318317
std::vector<Ort::Value> inputTensors;
319318
std::vector<Ort::Value> outputTensors;

c_cxx/OpenVINO_EP/Linux/squeezenet_classification/squeezenet_cpp_app_io.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,8 @@ int main(int argc, char* argv[])
260260
std::cout << "Number of Input Nodes: " << numInputNodes << std::endl;
261261
std::cout << "Number of Output Nodes: " << numOutputNodes << std::endl;
262262

263-
const char* inputName = session.GetInputName(0, allocator);
263+
auto inputNodeName = session.GetInputNameAllocated(0, allocator);
264+
const char* inputName = inputNodeName.get();
264265
std::cout << "Input Name: " << inputName << std::endl;
265266

266267
Ort::TypeInfo inputTypeInfo = session.GetInputTypeInfo(0);
@@ -272,7 +273,8 @@ int main(int argc, char* argv[])
272273
std::vector<int64_t> inputDims = inputTensorInfo.GetShape();
273274
std::cout << "Input Dimensions: " << inputDims << std::endl;
274275

275-
const char* outputName = session.GetOutputName(0, allocator);
276+
auto outputNodeName = session.GetOutputNameAllocated(0, allocator);
277+
const char* outputName = outputNodeName.get();
276278
std::cout << "Output Name: " << outputName << std::endl;
277279

278280
Ort::TypeInfo outputTypeInfo = session.GetOutputTypeInfo(0);

c_cxx/OpenVINO_EP/Windows/squeezenet_classification/squeezenet_cpp_app.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,8 @@ int main(int argc, char* argv[])
235235
std::cout << "Number of Input Nodes: " << numInputNodes << std::endl;
236236
std::cout << "Number of Output Nodes: " << numOutputNodes << std::endl;
237237

238-
const char* inputName = session.GetInputName(0, allocator);
238+
auto inputNodeName = session.GetInputNameAllocated(0, allocator);
239+
const char* inputName = inputNodeName.get();
239240
std::cout << "Input Name: " << inputName << std::endl;
240241

241242
Ort::TypeInfo inputTypeInfo = session.GetInputTypeInfo(0);
@@ -247,7 +248,8 @@ int main(int argc, char* argv[])
247248
std::vector<int64_t> inputDims = inputTensorInfo.GetShape();
248249
std::cout << "Input Dimensions: " << inputDims << std::endl;
249250

250-
const char* outputName = session.GetOutputName(0, allocator);
251+
auto outputNodeName = session.GetOutputNameAllocated(0, allocator);
252+
const char* outputName = outputNodeName.get();
251253
std::cout << "Output Name: " << outputName << std::endl;
252254

253255
Ort::TypeInfo outputTypeInfo = session.GetOutputTypeInfo(0);

0 commit comments

Comments
 (0)