HDK
|
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 |
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.
|
inlineexplicit |
Definition at line 1423 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1456 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1438 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1426 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1444 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1450 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1432 of file onnxruntime_cxx_inline.h.