tvm::te::TensorNode Class Reference

Node to represent 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 final
	Get the shape of the result. More...
PrimType	GetDataType () const final
	Get the raw element dtype of the result. More...
PrimExpr	ToPrimExpr () const final
ffi::String	GetNameHint () const final
	Get the name hint of the data producer. More...
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.Tensor", TensorNode, DataProducerNode)
Public Member Functions inherited from tvm::tirx::DataProducerNode
virtual	~DataProducerNode ()
	destructor. More...
	TVM_FFI_DECLARE_OBJECT_INFO ("tirx.DataProducer", DataProducerNode, PrimExprConvertibleNode)
Public Member Functions inherited from tvm::PrimExprConvertibleNode
virtual	~PrimExprConvertibleNode ()
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExprConvertible", PrimExprConvertibleNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Array< PrimExpr >	shape
	The shape of the tensor. More...
PrimType	dtype = PrimType::Void()
	dtype in the content of the tensor More...
Operation	op
	the source operation, can be None More...
int	value_index {0}
	the output index from source operation More...

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

Detailed Description

Node to represent a tensor.

Member Function Documentation

GetDataType()

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

inlinefinalvirtual

Get the raw element dtype of the result.

Returns

The raw dtype.

Implements tvm::tirx::DataProducerNode.

GetNameHint()

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

finalvirtual

Get the name hint of the data producer.

Returns

The data type.

Implements tvm::tirx::DataProducerNode.

GetShape()

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

inlinefinalvirtual

Get the shape of the result.

Returns

The shape.

Implements tvm::tirx::DataProducerNode.

RegisterReflection()

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

static

ToPrimExpr()

PrimExpr tvm::te::TensorNode::ToPrimExpr ( ) const

finalvirtual

Implements tvm::PrimExprConvertibleNode.

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

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

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