diff --git a/pkg/onnx/onnx.cpp b/pkg/onnx/onnx.cpp index dfc9420..a84e72d 100644 --- a/pkg/onnx/onnx.cpp +++ b/pkg/onnx/onnx.cpp @@ -245,7 +245,7 @@ extern "C" int onnx_get_input_shape( return -1; } - OrtTensorTypeAndShapeInfo* tensor_info = nullptr; + const OrtTensorTypeAndShapeInfo* tensor_info = nullptr; status = g_ort->CastTypeInfoToTensorInfo(type_info, &tensor_info); if (status) { last_error = g_ort->GetErrorMessage(status); @@ -263,7 +263,7 @@ extern "C" int onnx_get_input_shape( g_ort->GetDimensions(tensor_info, dims, dim_count); - g_ort->ReleaseTensorTypeAndShapeInfo(tensor_info); + g_ort->ReleaseTensorTypeAndShapeInfo((OrtTensorTypeAndShapeInfo*)tensor_info); g_ort->ReleaseTypeInfo(type_info); return (int)dim_count; @@ -289,7 +289,7 @@ extern "C" int onnx_get_output_shape( return -1; } - OrtTensorTypeAndShapeInfo* tensor_info = nullptr; + const OrtTensorTypeAndShapeInfo* tensor_info = nullptr; status = g_ort->CastTypeInfoToTensorInfo(type_info, &tensor_info); if (status) { last_error = g_ort->GetErrorMessage(status); @@ -307,7 +307,7 @@ extern "C" int onnx_get_output_shape( g_ort->GetDimensions(tensor_info, dims, dim_count); - g_ort->ReleaseTensorTypeAndShapeInfo(tensor_info); + g_ort->ReleaseTensorTypeAndShapeInfo((OrtTensorTypeAndShapeInfo*)tensor_info); g_ort->ReleaseTypeInfo(type_info); return (int)dim_count;