tvm::te::Operation Class Reference

Operation that produces tensors. More...

#include <tensor.h>

Inheritance diagram for tvm::te::Operation:

Collaboration diagram for tvm::te::Operation:

Public Types
using	ContainerType = OperationNode
	specify container node More...

Public Member Functions
	Operation ()
	default constructor More...
	Operation (ObjectPtr< Object > n)
const OperationNode *	operator-> () const
	access the internal node container More...
Tensor	output (size_t i) const
	get the i-th output of the operation. More...

Detailed Description

Operation that produces tensors.

Member Typedef Documentation

ContainerType

using tvm::te::Operation::ContainerType = OperationNode

specify container node

Constructor & Destructor Documentation

Operation() [1/2]

tvm::te::Operation::Operation ( )

inline

default constructor

Operation() [2/2]

tvm::te::Operation::Operation ( ObjectPtr< Object >  n )

inlineexplicit

Member Function Documentation

operator->()

const OperationNode * tvm::te::Operation::operator-> ( ) const

inline

access the internal node container

Returns

the pointer to the internal node container

output()

Tensor tvm::te::Operation::output

(

size_t

i

)

const

get the i-th output of the operation.

Parameters

i	the output index.

Returns

The i-th output.

---

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

• include/tvm/te/tensor.h
• include/tvm/te/operation.h