HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Member Functions | List of all members
Ort::KernelContext Struct Reference

This class wraps a raw pointer OrtKernelContext* that is being passed to the custom kernel Compute() method. Use it to safely access context attributes, input and output parameters with exception safety guarantees. See usage example in onnxruntime/test/testdata/custom_op_library/custom_op_library.cc More...

#include <onnxruntime_cxx_api.h>

Public Member Functions

 KernelContext (OrtKernelContext *context)
 
size_t GetInputCount () const
 
size_t GetOutputCount () const
 
ConstValue GetInput (size_t index) const
 
UnownedValue GetOutput (size_t index, const int64_t *dim_values, size_t dim_count) const
 
UnownedValue GetOutput (size_t index, const std::vector< int64_t > &dims) const
 
void * GetGPUComputeStream () const
 

Detailed Description

This class wraps a raw pointer OrtKernelContext* that is being passed to the custom kernel Compute() method. Use it to safely access context attributes, input and output parameters with exception safety guarantees. See usage example in onnxruntime/test/testdata/custom_op_library/custom_op_library.cc

Definition at line 1498 of file onnxruntime_cxx_api.h.

Constructor & Destructor Documentation

Ort::KernelContext::KernelContext ( OrtKernelContext *  context)
inlineexplicit

Definition at line 1423 of file onnxruntime_cxx_inline.h.

Member Function Documentation

void * Ort::KernelContext::GetGPUComputeStream ( ) const
inline

Definition at line 1456 of file onnxruntime_cxx_inline.h.

ConstValue Ort::KernelContext::GetInput ( size_t  index) const
inline

Definition at line 1438 of file onnxruntime_cxx_inline.h.

size_t Ort::KernelContext::GetInputCount ( ) const
inline

Definition at line 1426 of file onnxruntime_cxx_inline.h.

UnownedValue Ort::KernelContext::GetOutput ( size_t  index,
const int64_t *  dim_values,
size_t  dim_count 
) const
inline

Definition at line 1444 of file onnxruntime_cxx_inline.h.

UnownedValue Ort::KernelContext::GetOutput ( size_t  index,
const std::vector< int64_t > &  dims 
) const
inline

Definition at line 1450 of file onnxruntime_cxx_inline.h.

size_t Ort::KernelContext::GetOutputCount ( ) const
inline

Definition at line 1432 of file onnxruntime_cxx_inline.h.

The documentation for this struct was generated from the following files: