tvm::te::OperationNode Class Reference

Base class of all operation nodes. More...

#include <operation.h>

Inheritance diagram for tvm::te::OperationNode:

Collaboration diagram for tvm::te::OperationNode:

Public Member Functions
virtual	~OperationNode ()
virtual int	num_outputs () const =0
virtual DataType	output_dtype (size_t i) const =0
	Get data type. i-th output tensor. More...
virtual ffi::Array< PrimExpr >	output_shape (size_t i) const =0
	Get shape of i-th output tensor. More...
virtual ffi::Array< Tensor >	InputTensors () const =0
	List all the input Tensors. More...
	TVM_FFI_DECLARE_OBJECT_INFO ("te.Operation", OperationNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
std::string	name
	optional name of the operation More...
std::string	tag
	optional tag of the operation More...
ffi::Map< ffi::String, ffi::Any >	attrs
	additional attributes of the operation More...

Detailed Description

Base class of all operation nodes.

Constructor & Destructor Documentation

~OperationNode()

virtual tvm::te::OperationNode::~OperationNode ( )

inlinevirtual

Member Function Documentation

InputTensors()

virtual ffi::Array<Tensor> tvm::te::OperationNode::InputTensors ( ) const

pure virtual

List all the input Tensors.

Returns

List of input tensors.

Implemented in tvm::te::ExternOpNode, tvm::te::ScanOpNode, tvm::te::ComputeOpNode, and tvm::te::PlaceholderOpNode.

num_outputs()

virtual int tvm::te::OperationNode::num_outputs ( ) const

pure virtual

Returns

number of outputs

Implemented in tvm::te::ExternOpNode, tvm::te::ScanOpNode, tvm::te::ComputeOpNode, and tvm::te::PlaceholderOpNode.

output_dtype()

virtual DataType tvm::te::OperationNode::output_dtype ( size_t  i ) const

pure virtual

Get data type. i-th output tensor.

Parameters

i	The output index.

Returns

type of i-th output.

Implemented in tvm::te::ExternOpNode, tvm::te::ScanOpNode, tvm::te::ComputeOpNode, and tvm::te::PlaceholderOpNode.

output_shape()

virtual ffi::Array<PrimExpr> tvm::te::OperationNode::output_shape ( size_t  i ) const

pure virtual

Get shape of i-th output tensor.

Parameters

i	The output index.

Returns

shape of i-th output.

Implemented in tvm::te::BaseComputeOpNode, tvm::te::ExternOpNode, tvm::te::ScanOpNode, and tvm::te::PlaceholderOpNode.

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO()

tvm::te::OperationNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"te.Operation"	,
		OperationNode	,
		Object
	)

Member Data Documentation

attrs

ffi::Map<ffi::String, ffi::Any> tvm::te::OperationNode::attrs

additional attributes of the operation

name

std::string tvm::te::OperationNode::name

optional name of the operation

tag

std::string tvm::te::OperationNode::tag

optional tag of the operation

---

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

• include/tvm/te/operation.h