Fix C-API sample (causing internal build failure (#1047)

This commit is contained in:
jignparm 2019-05-16 15:49:39 -07:00 committed by GitHub
parent 99556b111d
commit 11069765dc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -78,7 +78,7 @@ int main(int argc, char* argv[]) {
printf("Input %zu : type=%d\n", i, type);
// print input shapes/dims
size_t num_dims = OrtGetNumOfDimensions(tensor_info);
size_t num_dims = OrtGetDimensionsCount(tensor_info);
printf("Input %zu : num_dims=%zu\n", i, num_dims);
input_node_dims.resize(num_dims);
OrtGetDimensions(tensor_info, (int64_t*)input_node_dims.data(), num_dims);