saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix C-API sample. Update Issue template. (#750)

Browse source 
* Fix C-API sample. Update Issue template.

* switch back to signed int

* update from int to size_t
This commit is contained in:
jignparm 2019-04-02 13:37:50 -07:00 committed by GitHub
parent afe3aae29f
commit acc8ac58d2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.github/ISSUE_TEMPLATE/bug-performance-issue.md vendored
View file

@ -12,6 +12,7 @@ A clear and concise description of what the bug is.
- ONNX Runtime installed from (source or binary):
- ONNX Runtime version:
- Python version:
- Visual Studio version (if applicable):
- GCC/Compiler version (if compiling from source):
- CUDA/cuDNN version:
- GPU model and memory:

10
csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests.Capi/C_Api_Sample.cpp
View file

@ -57,13 +57,13 @@ int main(int argc, char* argv[]) {
// print number of model input nodes
status = OrtSessionGetInputCount(session, &num_input_nodes);
std::vector<const char*> input_node_names(num_input_nodes);
std::vector<size_t> input_node_dims; // simplify... this model has only 1 input node {1, 3, 224, 224}.
std::vector<int64_t> input_node_dims; // simplify... this model has only 1 input node {1, 3, 224, 224}.
// Otherwise need vector<vector<>>
printf("Number of inputs = %zu\n", num_input_nodes);
// iterate over all input nodes
for (int i = 0; i < num_input_nodes; i++) {
for (size_t i = 0; i < num_input_nodes; i++) {
// print input node names
char* input_name;
status = OrtSessionGetInputName(session, i, allocator, &input_name);
@ -82,7 +82,7 @@ int main(int argc, char* argv[]) {
printf("Input %d : num_dims=%zu\n", i, num_dims);
input_node_dims.resize(num_dims);
OrtGetDimensions(tensor_info, (int64_t*)input_node_dims.data(), num_dims);
for (int j = 0; j < num_dims; j++)
for (size_t j = 0; j < num_dims; j++)
printf("Input %d : dim %d=%jd\n", i, j, input_node_dims[j]);
OrtReleaseTypeInfo(typeinfo);
@ -114,7 +114,7 @@ int main(int argc, char* argv[]) {
std::vector<const char*> output_node_names = {"softmaxout_1"};
// initialize input data with values in [0.0, 1.0]
for (unsigned int i = 0; i < input_tensor_size; i++)
for (size_t i = 0; i < input_tensor_size; i++)
input_tensor_values[i] = (float)i / (input_tensor_size + 1);
// create input tensor object from data values
@ -152,4 +152,4 @@ int main(int argc, char* argv[]) {
OrtReleaseEnv(env);
printf("Done!\n");
return 0;
}
}