tvm::te::ExternOpNode Class Reference

External computation that cannot be splitted. More...

#include <operation.h>

Inheritance diagram for tvm::te::ExternOpNode:

Collaboration diagram for tvm::te::ExternOpNode:

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

Static Public Member Functions
static void	RegisterReflection ()
Static Public Member Functions inherited from tvm::te::OperationNode
static void	RegisterReflection ()

Public Attributes
ffi::Array< Tensor >	inputs
	The input tensors. More...
ffi::Array< Buffer >	input_placeholders
	Symbolic placeholder representation of inputs. More...
ffi::Array< Buffer >	output_placeholders
	Symbolic placeholder representation of outputs. More...
Stmt	body
	the statement that generates the computation. More...
Public Attributes inherited from tvm::te::OperationNode
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

External computation that cannot be splitted.

Constructor & Destructor Documentation

ExternOpNode()

tvm::te::ExternOpNode::ExternOpNode ( )

inline

constructor

Member Function Documentation

InputTensors()

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

finalvirtual

List all the input Tensors.

Returns

List of input tensors.

Implements tvm::te::OperationNode.

num_outputs()

int tvm::te::ExternOpNode::num_outputs ( ) const

finalvirtual

Returns

number of outputs

Implements tvm::te::OperationNode.

output_dtype()

DataType tvm::te::ExternOpNode::output_dtype ( size_t  i ) const

finalvirtual

Get data type. i-th output tensor.

Parameters

i	The output index.

Returns

type of i-th output.

Implements tvm::te::OperationNode.

output_shape()

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

finalvirtual

Get shape of i-th output tensor.

Parameters

i	The output index.

Returns

shape of i-th output.

Implements tvm::te::OperationNode.

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::te::ExternOpNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"te.ExternOp"	,
		ExternOpNode	,
		OperationNode
	)

Member Data Documentation

body

Stmt tvm::te::ExternOpNode::body

the statement that generates the computation.

input_placeholders

ffi::Array<Buffer> tvm::te::ExternOpNode::input_placeholders

Symbolic placeholder representation of inputs.

inputs

ffi::Array<Tensor> tvm::te::ExternOpNode::inputs

The input tensors.

output_placeholders

ffi::Array<Buffer> tvm::te::ExternOpNode::output_placeholders

Symbolic placeholder representation of outputs.

---

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

• include/tvm/te/operation.h