Managed reference to TensorStructInfoNode. More...

#include <struct_info.h>

Inheritance diagram for tvm::relax::TensorStructInfo:

Collaboration diagram for tvm::relax::TensorStructInfo:

Public Member Functions
	TensorStructInfo (Expr shape, DataType dtype, Optional< VDevice > vdevice=NullOpt, Span span=Span())
	Construction with a known shape expression. More...

	TensorStructInfo (DataType dtype, int ndim, Optional< VDevice > vdevice=NullOpt, Span span=Span())
	Construction with an unknown shape expression. More...

	TVM_DEFINE_OBJECT_REF_METHODS (TensorStructInfo, StructInfo, TensorStructInfoNode)

Public Member Functions inherited from tvm::relax::StructInfo
	TVM_DEFINE_OBJECT_REF_METHODS (StructInfo, ObjectRef, StructInfoNode)

Public Member Functions inherited from tvm::runtime::ObjectRef
	ObjectRef ()=default
	default constructor More...

	ObjectRef (ObjectPtr< Object > data)
	Constructor from existing object ptr. More...

bool	same_as (const ObjectRef &other) const
	Comparator. More...

bool	operator== (const ObjectRef &other) const
	Comparator. More...

bool	operator!= (const ObjectRef &other) const
	Comparator. More...

bool	operator< (const ObjectRef &other) const
	Comparator. More...

bool	defined () const

const Object *	get () const

const Object *	operator-> () const

bool	unique () const

int	use_count () const

template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType *	as () const
	Try to downcast the internal Object to a raw pointer of a corresponding type. More...

template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType >	as () const
	Try to downcast the ObjectRef to a Optional<T> of the requested type. More...

Additional Inherited Members
Public Types inherited from tvm::runtime::ObjectRef
using	ContainerType = Object
	type indicate the container type. More...

Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool	_type_is_nullable = true

Protected Member Functions inherited from tvm::runtime::ObjectRef
Object *	get_mutable () const

Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T	DowncastNoCheck (ObjectRef ref)
	Internal helper function downcast a ref without check. More...

static void	FFIClearAfterMove (ObjectRef *ref)
	Clear the object ref data field without DecRef after we successfully moved the field. More...

template<typename ObjectType >
static ObjectPtr< ObjectType >	GetDataPtr (const ObjectRef &ref)
	Internal helper function get data_ as ObjectPtr of ObjectType. More...

Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Object >	data_
	Internal pointer that backs the reference. More...

Detailed Description

Managed reference to TensorStructInfoNode.

See also: TensorStructInfoNode

Constructor & Destructor Documentation

◆ TensorStructInfo() [1/2]

tvm::relax::TensorStructInfo::TensorStructInfo	(	Expr	shape,
		DataType	dtype,
		Optional< VDevice >	vdevice = `NullOpt`,
		Span	span = `Span()`
	)

Construction with a known shape expression.

Parameters

shape	The shape of the tensor.
dtype	The data type of tensor's elements.
vdevice	The virtual device.
span	The span of the AST.

Note: shape must already be normalized.

◆ TensorStructInfo() [2/2]

tvm::relax::TensorStructInfo::TensorStructInfo	(	DataType	dtype,
		int	ndim,
		Optional< VDevice >	vdevice = `NullOpt`,
		Span	span = `Span()`
	)

Construction with an unknown shape expression.

Parameters

dtype	The data type of tensor's elements.
ndim	The number of dimensions
vdevice	The virtual device.
span	The span of the AST.

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::relax::TensorStructInfo::TVM_DEFINE_OBJECT_REF_METHODS	(	TensorStructInfo	,
		StructInfo	,
		TensorStructInfoNode
	)

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

include/tvm/relax/struct_info.h