Common base-class for all sparse-tensors (with different element types). More...

#include <data_types.h>

Inheritance diagram for onnxruntime::SparseTensorTypeBase:

Public Member Functions
bool	IsCompatible (const ONNX_NAMESPACE::TypeProto &type_proto) const override
	this API will be used to check type compatibility at runtime More...

DeleteFunc	GetDeleteFunc () const override

const ONNX_NAMESPACE::TypeProto *	GetTypeProto () const override
	Retrieves an instance of TypeProto for a given MLDataType. More...

virtual MLDataType	GetElementType () const

	ORT_DISALLOW_COPY_ASSIGNMENT_AND_MOVE (SparseTensorTypeBase)

Public Member Functions inherited from onnxruntime::DataTypeImpl
virtual	~DataTypeImpl ()=default

size_t	Size () const

bool	IsTensorType () const

bool	IsTensorSequenceType () const

bool	IsSparseTensorType () const

bool	IsOptionalType () const

bool	IsNonTensorType () const

bool	IsPrimitiveDataType () const

const TensorTypeBase *	AsTensorType () const

const SequenceTensorTypeBase *	AsSequenceTensorType () const

const SparseTensorTypeBase *	AsSparseTensorType () const

const OptionalTypeBase *	AsOptionalType () const

const NonTensorTypeBase *	AsNonTensorType () const

const PrimitiveDataTypeBase *	AsPrimitiveDataType () const

Static Public Member Functions
static MLDataType	Type ()

Static Public Member Functions inherited from onnxruntime::DataTypeImpl
template<typename T >
static MLDataType	GetType ()

template<typename elemT >
static MLDataType	GetTensorType ()

template<typename elemT >
static MLDataType	GetSequenceTensorType ()

template<typename elemT >
static MLDataType	GetSparseTensorType ()

template<typename T , typename elemT >
static MLDataType	GetOptionalType ()

static MLDataType	TypeFromProto (const ONNX_NAMESPACE::TypeProto &proto)

static const TensorTypeBase *	TensorTypeFromONNXEnum (int type)

static const SequenceTensorTypeBase *	SequenceTensorTypeFromONNXEnum (int type)

static const SparseTensorTypeBase *	SparseTensorTypeFromONNXEnum (int type)

static const char *	ToString (MLDataType type)

static std::vector< std::string >	ToString (const std::vector< MLDataType > &types)

static void	RegisterDataType (MLDataType)

static MLDataType	GetDataType (const std::string &)

static const std::vector < MLDataType > &	AllTensorTypes ()

static const std::vector < MLDataType > &	AllFixedSizeTensorTypes ()

static const std::vector < MLDataType > &	AllSequenceTensorTypes ()

static const std::vector < MLDataType > &	AllFixedSizeSequenceTensorTypes ()

static const std::vector < MLDataType > &	AllNumericTensorTypes ()

static const std::vector < MLDataType > &	AllIEEEFloatTensorTypes ()

static const std::vector < MLDataType > &	AllFixedSizeTensorExceptHalfTypes ()

static const std::vector < MLDataType > &	AllIEEEFloatTensorExceptHalfTypes ()

static const std::vector < MLDataType > &	AllTensorAndSequenceTensorTypes ()

static const std::vector < MLDataType > &	AllFixedSizeTensorAndSequenceTensorTypes ()

static const std::vector < MLDataType > &	AllOptionalTypes ()

static const std::vector < MLDataType > &	AllTensorAndSequenceTensorAndOptionalTypes ()

Protected Member Functions
ONNX_NAMESPACE::TypeProto &	MutableTypeProto ()

	SparseTensorTypeBase ()

	~SparseTensorTypeBase () override

Protected Member Functions inherited from onnxruntime::DataTypeImpl
	DataTypeImpl (GeneralType type, size_t size)

Additional Inherited Members
Public Types inherited from onnxruntime::DataTypeImpl
enum	GeneralType { GeneralType::kInvalid = 0, GeneralType::kNonTensor = 1, GeneralType::kTensor = 2, GeneralType::kTensorSequence = 3, GeneralType::kSparseTensor = 4, GeneralType::kOptional = 5, GeneralType::kPrimitive = 6 }

Public Attributes inherited from onnxruntime::DataTypeImpl
const GeneralType	type_

const size_t	size_

Detailed Description

Common base-class for all sparse-tensors (with different element types).

Definition at line 501 of file data_types.h.

Constructor & Destructor Documentation

onnxruntime::SparseTensorTypeBase::SparseTensorTypeBase ( )

protected

onnxruntime::SparseTensorTypeBase::~SparseTensorTypeBase ( )

overrideprotected

Member Function Documentation

DeleteFunc onnxruntime::SparseTensorTypeBase::GetDeleteFunc ( ) const

overridevirtual

Implements onnxruntime::DataTypeImpl.

virtual MLDataType onnxruntime::SparseTensorTypeBase::GetElementType ( ) const

inlinevirtual

Reimplemented in onnxruntime::SparseTensorType< elemT >.

Definition at line 511 of file data_types.h.

const ONNX_NAMESPACE::TypeProto* onnxruntime::SparseTensorTypeBase::GetTypeProto ( ) const

overridevirtual

Retrieves an instance of TypeProto for a given MLDataType.

Returns: optional TypeProto. Only ONNX types has type proto, non-ONNX types will return nullptr.

Implements onnxruntime::DataTypeImpl.

bool onnxruntime::SparseTensorTypeBase::IsCompatible ( const ONNX_NAMESPACE::TypeProto & type_proto ) const

overridevirtual

this API will be used to check type compatibility at runtime

Parameters

type_proto a TypeProto instance that is constructed for a specific type will be checked against a TypeProto instance contained within a corresponding MLDataType instance.

Implements onnxruntime::DataTypeImpl.

ONNX_NAMESPACE::TypeProto& onnxruntime::SparseTensorTypeBase::MutableTypeProto ( )

protected

onnxruntime::SparseTensorTypeBase::ORT_DISALLOW_COPY_ASSIGNMENT_AND_MOVE ( SparseTensorTypeBase )

static MLDataType onnxruntime::SparseTensorTypeBase::Type ( )

static

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

onnxruntime/core/framework/data_types.h

Public Member Functions

Static Public Member Functions

Protected Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation