#include <op_kernel_info.h>

Inheritance diagram for onnxruntime::OpKernelInfo:

Public Member Functions
	OpKernelInfo (const onnxruntime::Node &node, const KernelDef &kernel_def, const IExecutionProvider &execution_provider, const std::unordered_map< int, OrtValue > &constant_initialized_tensors, const OrtValueNameIdxMap &mlvalue_name_idx_map, const DataTransferManager &data_transfer_mgr)

	OpKernelInfo (const OpKernelInfo &other)

const OrtMemoryInfo &	GetMemoryInfo (int device_id, OrtMemType mem_type) const

AllocatorPtr	GetAllocator (int device_id, OrtMemType mem_type) const

const KernelDef &	GetKernelDef () const

const IExecutionProvider *	GetExecutionProvider () const noexcept

const DataTransferManager &	GetDataTransferManager () const noexcept

const onnxruntime::Node &	node () const noexcept

bool	TryGetConstantInput (int input_index, const Tensor **constant_input_value) const

Public Member Functions inherited from onnxruntime::OpNodeProtoHelper< ProtoHelperNodeContext >
	OpNodeProtoHelper (const ProtoHelperNodeContext *impl)

MUST_USE_RESULT Status	GetAttr (const std::string &name, T *value) const

T	GetAttrOrDefault (const std::string &name, const T &default_value) const

void	GetAttrOrDefault (const std::string &name, T *value, const T &default_value) const

MUST_USE_RESULT std::vector< T >	GetAttrsOrDefault (const std::string &name, const std::vector< T > &default_value=std::vector< T >{}) const

MUST_USE_RESULT TensorShapeVector	GetAttrsOrDefault (const std::string &name, const TensorShapeVector &default_value=TensorShapeVector{}) const

MUST_USE_RESULT Status	GetAttrsAsSpan (const std::string &name, gsl::span< const T > &values) const
	Return a gsl::span that points to an array of primitive types held by AttributeProto This function allows to avoid copying big attributes locally into a kernel and operate on AttributeProto data directly. More...

MUST_USE_RESULT Status	GetAttrs (const std::string &name, TensorShapeVector &out) const

MUST_USE_RESULT Status	GetAttrs (const std::string &name, std::vector< T > &values) const

MUST_USE_RESULT Status	GetAttrs (const std::string &name, gsl::span< T > values) const

MUST_USE_RESULT Status	GetAttrsStringRefs (const std::string &name, std::vector< std::reference_wrapper< const std::string >> &refs) const

uint32_t	GetPrimitiveAttrElementCount (ONNX_NAMESPACE::AttributeProto_AttributeType type, const std::string &name) const noexcept

bool	HasPrimitiveAttribute (ONNX_NAMESPACE::AttributeProto_AttributeType type, const std::string &name) const noexcept

uint32_t	GetInputCount () const

uint32_t	GetOutputCount () const

const ONNX_NAMESPACE::TypeProto *	GetInputType (size_t index) const

const ONNX_NAMESPACE::TypeProto *	GetOutputType (size_t index) const

const ONNX_NAMESPACE::AttributeProto *	TryGetAttribute (const std::string &name) const

const ONNX_NAMESPACE::AttributeProto *	GetAttribute (const std::string &name) const

Detailed Description

Definition at line 23 of file op_kernel_info.h.

Constructor & Destructor Documentation

onnxruntime::OpKernelInfo::OpKernelInfo	(	const onnxruntime::Node &	node,
		const KernelDef &	kernel_def,
		const IExecutionProvider &	execution_provider,
		const std::unordered_map< int, OrtValue > &	constant_initialized_tensors,
		const OrtValueNameIdxMap &	mlvalue_name_idx_map,
		const DataTransferManager &	data_transfer_mgr
	)

explicit

onnxruntime::OpKernelInfo::OpKernelInfo ( const OpKernelInfo & other )

Member Function Documentation

AllocatorPtr onnxruntime::OpKernelInfo::GetAllocator	(	int	device_id,
		OrtMemType	mem_type
	)		const

const DataTransferManager& onnxruntime::OpKernelInfo::GetDataTransferManager ( ) const

noexcept

const IExecutionProvider* onnxruntime::OpKernelInfo::GetExecutionProvider ( ) const

noexcept

const KernelDef& onnxruntime::OpKernelInfo::GetKernelDef ( ) const

const OrtMemoryInfo& onnxruntime::OpKernelInfo::GetMemoryInfo	(	int	device_id,
		OrtMemType	mem_type
	)		const

const onnxruntime::Node& onnxruntime::OpKernelInfo::node ( ) const

noexcept

bool onnxruntime::OpKernelInfo::TryGetConstantInput	(	int	input_index,
		const Tensor **	constant_input_value
	)		const

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

onnxruntime/core/framework/op_kernel_info.h

Public Member Functions

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation