tvm::tir::DataProducerNode Class Reference

Base node for data producers. More...

#include <buffer.h>

Inheritance diagram for tvm::tir::DataProducerNode:

Collaboration diagram for tvm::tir::DataProducerNode:

Public Member Functions
virtual	~DataProducerNode ()
	destructor. More...
virtual ffi::Array< PrimExpr >	GetShape () const =0
	Get the shape of the result. More...
virtual DataType	GetDataType () const =0
	Get the data type of the result. More...
virtual ffi::String	GetNameHint () const =0
	Get the name hint of the data producer. More...
	TVM_FFI_DECLARE_OBJECT_INFO ("tir.DataProducer", DataProducerNode, PrimExprConvertibleNode)
Public Member Functions inherited from tvm::PrimExprConvertibleNode
virtual	~PrimExprConvertibleNode ()
virtual PrimExpr	ToPrimExpr () const =0
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExprConvertible", PrimExprConvertibleNode, Object)

Detailed Description

Base node for data producers.

A DataProducer stores necessary information(e.g. a tensor expression) to produce a multi-dimensional array. The stored information is opaque to the TIR. DataProducer can appear in high-level DSLs that are built on top of the TIR.

A valid TIR PrimFunc should not contain any DataProducer, high level DSLs should lower all DataProducers to Buffers before TIR transformations.

See also

tvm::te::Tensor

Constructor & Destructor Documentation

~DataProducerNode()

virtual tvm::tir::DataProducerNode::~DataProducerNode ( )

inlinevirtual

destructor.

Member Function Documentation

GetDataType()

virtual DataType tvm::tir::DataProducerNode::GetDataType ( ) const

pure virtual

Get the data type of the result.

Returns

The data type.

Implemented in tvm::te::TensorNode.

GetNameHint()

virtual ffi::String tvm::tir::DataProducerNode::GetNameHint ( ) const

pure virtual

Get the name hint of the data producer.

Returns

The data type.

Implemented in tvm::te::TensorNode.

GetShape()

virtual ffi::Array<PrimExpr> tvm::tir::DataProducerNode::GetShape ( ) const

pure virtual

Get the shape of the result.

Returns

The shape.

Implemented in tvm::te::TensorNode.

TVM_FFI_DECLARE_OBJECT_INFO()

tvm::tir::DataProducerNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"tir.DataProducer"	,
		DataProducerNode	,
		PrimExprConvertibleNode
	)

---

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

• include/tvm/tir/buffer.h