HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Types | Public Member Functions | List of all members
onnxruntime::OpKernel Class Referenceabstract

#include <op_kernel.h>

Public Types

using DoneCallback = std::function< void()>
 

Public Member Functions

 OpKernel (const OpKernelInfo &info)
 
virtual ~OpKernel ()=default
 
const onnxruntime::Node & Node () const
 
const onnxruntime::KernelDef & KernelDef () const
 
virtual Status Compute (_Inout_ OpKernelContext *context) const =0
 
virtual bool IsAsync () const
 
virtual Status ComputeAsync (_Inout_ OpKernelContext *, DoneCallback) const
 
virtual Status PrePack (const Tensor &, int, AllocatorPtr, bool &is_packed, PrePackedWeights *)
 
virtual Status UseSharedPrePackedBuffers (std::vector< BufferUniquePtr > &, int, bool &used_shared_buffers)
 
const OrtMemoryInfo & Allocator (int id, OrtMemType mem_type) const
 
const OpKernelInfo & Info () const
 

Detailed Description

Definition at line 43 of file op_kernel.h.

Member Typedef Documentation

using onnxruntime::OpKernel::DoneCallback = std::function<void()>

Definition at line 45 of file op_kernel.h.

Constructor & Destructor Documentation

onnxruntime::OpKernel::OpKernel ( const OpKernelInfo &  info)
inlineexplicit

Definition at line 47 of file op_kernel.h.

virtual onnxruntime::OpKernel::~OpKernel ( )
virtualdefault

Member Function Documentation

const OrtMemoryInfo& onnxruntime::OpKernel::Allocator ( int  id,
OrtMemType  mem_type 
) const
virtual Status onnxruntime::OpKernel::Compute ( _Inout_ OpKernelContext *  context) const
pure virtual
virtual Status onnxruntime::OpKernel::ComputeAsync ( _Inout_ OpKernelContext *  ,
DoneCallback   
) const
inlinevirtual

Definition at line 60 of file op_kernel.h.

const OpKernelInfo& onnxruntime::OpKernel::Info ( ) const
inline

Definition at line 125 of file op_kernel.h.

virtual bool onnxruntime::OpKernel::IsAsync ( ) const
inlinevirtual

Definition at line 55 of file op_kernel.h.

const onnxruntime::KernelDef& onnxruntime::OpKernel::KernelDef ( ) const
const onnxruntime::Node& onnxruntime::OpKernel::Node ( ) const
virtual Status onnxruntime::OpKernel::PrePack ( const Tensor &  ,
int  ,
AllocatorPtr  ,
bool &  is_packed,
PrePackedWeights *   
)
inlinevirtual

Definition at line 95 of file op_kernel.h.

virtual Status onnxruntime::OpKernel::UseSharedPrePackedBuffers ( std::vector< BufferUniquePtr > &  ,
int  ,
bool &  used_shared_buffers 
)
inlinevirtual

Definition at line 117 of file op_kernel.h.

The documentation for this class was generated from the following file: