#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	GetRemovableAttributes (InlinedVector< std::string > &removable_attributes) const

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

const OrtDevice	GetDevice (OrtMemType mem_type) const

const OpKernelInfo &	Info () const

Detailed Description

Definition at line 38 of file op_kernel.h.

Member Typedef Documentation

Definition at line 40 of file op_kernel.h.

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

inlineexplicit

Definition at line 42 of file op_kernel.h.

virtual onnxruntime::OpKernel::~OpKernel ( )

virtualdefault

virtual Status onnxruntime::OpKernel::Compute ( _Inout_ OpKernelContext * context ) const

pure virtual

virtual Status onnxruntime::OpKernel::ComputeAsync	(	_Inout_ OpKernelContext *	,
		DoneCallback
	)		const

inlinevirtual

Definition at line 55 of file op_kernel.h.

const OrtDevice onnxruntime::OpKernel::GetDevice ( OrtMemType mem_type ) const

virtual Status onnxruntime::OpKernel::GetRemovableAttributes ( InlinedVector< std::string > & removable_attributes ) const

inlinevirtual

Definition at line 102 of file op_kernel.h.

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

inline

Definition at line 133 of file op_kernel.h.

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

inlinevirtual

Definition at line 50 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 90 of file op_kernel.h.

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

inlinevirtual

Definition at line 125 of file op_kernel.h.

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