From 1ff29bfb3df23bfae97edcf857e06afdc1d08a91 Mon Sep 17 00:00:00 2001 From: Ryan Hill <38674843+RyanUnderhill@users.noreply.github.com> Date: Thu, 11 Apr 2019 10:41:07 -0700 Subject: [PATCH] Fix x86 calling convention break (#814) --- onnxruntime/core/session/custom_ops.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/session/custom_ops.cc b/onnxruntime/core/session/custom_ops.cc index 165bb13ecd..a821a718de 100644 --- a/onnxruntime/core/session/custom_ops.cc +++ b/onnxruntime/core/session/custom_ops.cc @@ -29,11 +29,11 @@ ORT_API_STATUS_IMPL(OrtKernelInfoGetAttribute_int64, _In_ const OrtKernelInfo* i return onnxruntime::ToOrtStatus(status); } -OrtValue* OrtKernelContext_GetInput(OrtKernelContext* context, _In_ size_t index) { +ORT_API(OrtValue*, OrtKernelContext_GetInput, OrtKernelContext* context, _In_ size_t index) { return reinterpret_cast(const_cast(reinterpret_cast(context)->GetInputMLValue(index))); }; -OrtValue* OrtKernelContext_GetOutput(OrtKernelContext* context, _In_ size_t index, _In_ const int64_t* dim_values, size_t dim_count) { +ORT_API(OrtValue*, OrtKernelContext_GetOutput, OrtKernelContext* context, _In_ size_t index, _In_ const int64_t* dim_values, size_t dim_count) { onnxruntime::TensorShape shape(dim_values, dim_count); return reinterpret_cast(reinterpret_cast(context)->OutputMLValue(index, shape)); };