HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
onnxruntime::TensorTypeBase Class Reference

All tensors base. More...

#include <data_types.h>

+ Inheritance diagram for onnxruntime::TensorTypeBase:

Public Member Functions

bool IsCompatible (const ONNX_NAMESPACE::TypeProto &type_proto) const override
 
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 (TensorTypeBase)
 
- 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 TensorTypeBaseAsTensorType () const
 
const SequenceTensorTypeBaseAsSequenceTensorType () const
 
const SparseTensorTypeBaseAsSparseTensorType () const
 
const OptionalTypeBaseAsOptionalType () const
 
const NonTensorTypeBaseAsNonTensorType () const
 
const PrimitiveDataTypeBaseAsPrimitiveDataType () 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 TensorTypeBaseTensorTypeFromONNXEnum (int type)
 
static const
SequenceTensorTypeBase
SequenceTensorTypeFromONNXEnum (int type)
 
static const SparseTensorTypeBaseSparseTensorTypeFromONNXEnum (int type)
 
static const char * ToString (MLDataType type)
 
static std::vector< std::stringToString (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 ()
 
 TensorTypeBase ()
 
 ~TensorTypeBase () 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

All tensors base.

Definition at line 395 of file data_types.h.

Constructor & Destructor Documentation

onnxruntime::TensorTypeBase::TensorTypeBase ( )
protected
onnxruntime::TensorTypeBase::~TensorTypeBase ( )
overrideprotected

Member Function Documentation

DeleteFunc onnxruntime::TensorTypeBase::GetDeleteFunc ( ) const
overridevirtual
virtual MLDataType onnxruntime::TensorTypeBase::GetElementType ( ) const
inlinevirtual

Reimplemented in onnxruntime::TensorType< elemT >.

Definition at line 408 of file data_types.h.

const ONNX_NAMESPACE::TypeProto* onnxruntime::TensorTypeBase::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::TensorTypeBase::IsCompatible ( const ONNX_NAMESPACE::TypeProto &  type_proto) const
overridevirtual

We first compare type_proto pointers and then if they do not match try to account for the case where TypeProto was created ad-hoc and not queried from MLDataType

Implements onnxruntime::DataTypeImpl.

ONNX_NAMESPACE::TypeProto& onnxruntime::TensorTypeBase::MutableTypeProto ( )
protected
onnxruntime::TensorTypeBase::ORT_DISALLOW_COPY_ASSIGNMENT_AND_MOVE ( TensorTypeBase  )
static MLDataType onnxruntime::TensorTypeBase::Type ( )
static

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