From aa93f2e23680924f0b83cce1ebec4fce6e579baa Mon Sep 17 00:00:00 2001 From: Baiju Meswani Date: Wed, 3 Mar 2021 20:38:17 -0800 Subject: [PATCH] move SetOutputMLValue from op_kernel.h to op_kernel_context.h --- include/onnxruntime/core/framework/op_kernel_context.h | 1 + 1 file changed, 1 insertion(+) diff --git a/include/onnxruntime/core/framework/op_kernel_context.h b/include/onnxruntime/core/framework/op_kernel_context.h index 833f6f9185..a87781b00c 100644 --- a/include/onnxruntime/core/framework/op_kernel_context.h +++ b/include/onnxruntime/core/framework/op_kernel_context.h @@ -175,6 +175,7 @@ class OpKernelContext { const OrtValue* GetInputMLValue(int index) const; const OrtValue* GetImplicitInputMLValue(int index) const; OrtValue* GetOutputMLValue(int index); + Status SetOutputMLValue(int index, const OrtValue& ort_value); // Creates the OrtValue* based on the shape, if it does not exist // The parameter nnz is used only for sparse-tensors and indicates the