tvm::relax::TensorStructInfo Class Reference

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=std::nullopt, Span span=Span())
	Construction with a known shape expression. More...
	TensorStructInfo (DataType dtype, int ndim, Optional< VDevice > vdevice=std::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)

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 = std::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 = std::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