saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-02 03:55:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix an SAL annotation in the C API

Browse source
This commit is contained in:
Changming Sun 2019-02-20 11:22:50 -08:00
parent fc7185f060
commit b02c1d80d4
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
include/onnxruntime/core/session/onnxruntime_c_api.h
View file

@ -284,7 +284,7 @@ ORT_API_STATUS(OrtCreateTensorAsOrtValue, _Inout_ OrtAllocator* allocator,
* \param out Should be freed by calling OrtReleaseValue
*/
ORT_API_STATUS(OrtCreateTensorWithDataAsOrtValue, _In_ const OrtAllocatorInfo* info,
_In_ void* p_data, size_t p_data_len, _In_ const size_t* shape, size_t shape_len,
_Inout_ void* p_data, size_t p_data_len, _In_ const size_t* shape, size_t shape_len,
ONNXTensorElementDataType type, _Out_ OrtValue** out);
// This function doesn't work with string tensor

2
onnxruntime/core/session/onnxruntime_c_api.cc
View file

@ -222,7 +222,7 @@ OrtStatus* CreateTensorImpl(const size_t* shape, size_t shape_len, const OrtAllo
* this function will create a copy of the allocator info
*/
ORT_API_STATUS_IMPL(OrtCreateTensorWithDataAsOrtValue, _In_ const OrtAllocatorInfo* info,
_In_ void* p_data, size_t p_data_len, _In_ const size_t* shape, size_t shape_len,
_Inout_ void* p_data, size_t p_data_len, _In_ const size_t* shape, size_t shape_len,
ONNXTensorElementDataType type, _Out_ OrtValue** out) {
API_IMPL_BEGIN
std::unique_ptr<Tensor> tensor;