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 Array< PrimExpr >	output_shape (size_t i) const =0
	Get shape of i-th output tensor. More...
virtual Array< Tensor >	InputTensors () const =0
	List all the input Tensors. More...
	TVM_DECLARE_BASE_OBJECT_INFO (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...
Map< String, ffi::Any >	attrs
	additional attributes of the operation More...

Static Public Attributes
static constexpr const char *	_type_key = "te.Operation"

Detailed Description

Base class of all operation nodes.

Constructor & Destructor Documentation

~OperationNode()

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

inlinevirtual

Member Function Documentation

InputTensors()

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

tvm::te::OperationNode::TVM_DECLARE_BASE_OBJECT_INFO	(	OperationNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::te::OperationNode::_type_key = "te.Operation"

staticconstexpr

attrs

Map<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