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...
DataType	GetDataType () const final
	Get the data type 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::tir::DataProducerNode
virtual	~DataProducerNode ()
	destructor. More...
	TVM_FFI_DECLARE_OBJECT_INFO ("tir.DataProducer", DataProducerNode, PrimExprConvertibleNode)
Public Member Functions inherited from tvm::PrimExprConvertibleNode
virtual	~PrimExprConvertibleNode ()
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExprConvertible", PrimExprConvertibleNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::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 TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

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()

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

finalvirtual

Get the name hint of the data producer.

Returns

The data type.

Implements tvm::tir::DataProducerNode.

GetShape()

ffi::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_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

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

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