Opaque construction-time node that represents a tensor. More...

#include <tensor.h>

Inheritance diagram for tvm::te::TensorNode:

Collaboration diagram for tvm::te::TensorNode:

Public Member Functions
ffi::Array< PrimExpr >	GetShape () const

PrimType	GetDataType () const

ffi::String	GetNameHint () const

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.Tensor", TensorNode, OpaqueExprNode)

Public Member Functions inherited from tvm::OpaqueExprNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.OpaqueExpr", OpaqueExprNode, ExprNode)

Public Member Functions inherited from tvm::ExprNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Expr", ExprNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::OpaqueExprNode
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::ExprNode
static void	RegisterReflection ()

Public Attributes
ffi::Array< PrimExpr >	shape
	The shape of the tensor.

PrimType	dtype = PrimType::Void()
	dtype in the content of the tensor

Operation	op
	the source operation, can be None

int	value_index {0}
	the output index from source operation

Public Attributes inherited from tvm::ExprNode
Span	span
	Span that points to the original source code. Reserved debug information.

Type	ty = Type::Missing()
	The deduced or annotated type of the expression.

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

Static Public Attributes inherited from tvm::OpaqueExprNode
static constexpr const uint32_t	_type_child_slots = 2

Static Public Attributes inherited from tvm::ExprNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

static constexpr const uint32_t	_type_child_slots = 64

Detailed Description

Opaque construction-time node that represents a tensor.

Member Function Documentation

◆ GetDataType()

PrimType tvm::te::TensorNode::GetDataType ( ) const

inline

◆ GetNameHint()

ffi::String tvm::te::TensorNode::GetNameHint ( ) const

◆ GetShape()

ffi::Array< PrimExpr > tvm::te::TensorNode::GetShape ( ) const

inline

◆ RegisterReflection()

static void tvm::te::TensorNode::RegisterReflection ( )

static

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::te::TensorNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"te.Tensor"	,
		TensorNode	,
		OpaqueExprNode
	)

Member Data Documentation

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::te::TensorNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

staticconstexpr

◆ dtype

PrimType tvm::te::TensorNode::dtype = PrimType::Void()

dtype in the content of the tensor

◆ op

Operation tvm::te::TensorNode::op

the source operation, can be None

◆ shape

ffi::Array<PrimExpr> tvm::te::TensorNode::shape

The shape of the tensor.

◆ value_index

int tvm::te::TensorNode::value_index {0}

the output index from source operation

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

include/tvm/te/tensor.h

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes