#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, const AllocatorMap &allocators, const ConfigOptions &config_options)

	OpKernelInfo (const OpKernelInfo &other)

const OrtDevice	GetDevice (OrtMemType mem_type) const

AllocatorPtr	GetAllocator (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

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

const AllocatorMap &	GetAllocators () const

const ConfigOptions &	GetConfigOptions () const

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

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

T	GetAttr (const std::string &name) 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

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

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

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...

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

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

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

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,
		const AllocatorMap &	allocators,
		const ConfigOptions &	config_options
	)

explicit

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

Member Function Documentation

AllocatorPtr onnxruntime::OpKernelInfo::GetAllocator ( OrtMemType mem_type ) const

const AllocatorMap& onnxruntime::OpKernelInfo::GetAllocators ( ) const

inline

Definition at line 52 of file op_kernel_info.h.

const ConfigOptions& onnxruntime::OpKernelInfo::GetConfigOptions ( ) const

inline

Definition at line 54 of file op_kernel_info.h.

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

noexcept

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

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

noexcept

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

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

noexcept

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

bool onnxruntime::OpKernelInfo::TryGetConstantInput	(	int	input_index,
		const OrtValue **	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