saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-22 22:01:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

move SetOutputMLValue from op_kernel.h to op_kernel_context.h

Browse source
This commit is contained in:
Baiju Meswani 2021-03-03 20:38:17 -08:00
parent d5667554e6
commit aa93f2e236
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
include/onnxruntime/core/framework/op_kernel_context.h
View file

@ -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