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
Array< PrimExpr >	GetShape () const final
	Get the shape of the result. More...
DataType	GetDataType () const final
	Get the data type of the result. More...
PrimExpr	ToPrimExpr () const final
String	GetNameHint () const final
	Get the name hint of the data producer. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (TensorNode, DataProducerNode)
Public Member Functions inherited from tvm::tir::DataProducerNode
virtual	~DataProducerNode ()
	destructor. More...
	TVM_DECLARE_BASE_OBJECT_INFO (DataProducerNode, PrimExprConvertibleNode)
Public Member Functions inherited from tvm::PrimExprConvertibleNode
virtual	~PrimExprConvertibleNode ()
	TVM_DECLARE_BASE_OBJECT_INFO (PrimExprConvertibleNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Array< PrimExpr >	shape
	The shape of the tensor. More...
DataType	dtype
	data type 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 const char *	_type_key = "te.Tensor"
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode
Static Public Attributes inherited from tvm::tir::DataProducerNode
static constexpr const char *	_type_key = "tir.DataProducer"
Static Public Attributes inherited from tvm::PrimExprConvertibleNode
static constexpr const char *	_type_key = "ir.PrimExprConvertible"

Detailed Description

Node to represent a tensor.

Member Function Documentation

GetDataType()

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

inlinefinalvirtual

Get the data type of the result.

Returns

The data type.

Implements tvm::tir::DataProducerNode.

GetNameHint()

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

finalvirtual

Get the name hint of the data producer.

Returns

The data type.

Implements tvm::tir::DataProducerNode.

GetShape()

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

inlinefinalvirtual

Get the shape of the result.

Returns

The shape.

Implements tvm::tir::DataProducerNode.

RegisterReflection()

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

static

ToPrimExpr()

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

finalvirtual

Implements tvm::PrimExprConvertibleNode.

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::te::TensorNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	TensorNode	,
		DataProducerNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::te::TensorNode::_type_key = "te.Tensor"

staticconstexpr

_type_s_eq_hash_kind

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

staticconstexpr

dtype

DataType tvm::te::TensorNode::dtype

data type in the content of the tensor

op

Operation tvm::te::TensorNode::op

the source operation, can be None

shape

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