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 Types inherited from tvm::runtime::ObjectRef
using	ContainerType = Object
	type indicate the container type. 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...
Public Member Functions inherited from tvm::runtime::ObjectRef
	ObjectRef ()=default
	default constructor More...
	ObjectRef (ObjectPtr< Object > data)
	Constructor from existing object ptr. More...
bool	same_as (const ObjectRef &other) const
	Comparator. More...
bool	operator== (const ObjectRef &other) const
	Comparator. More...
bool	operator!= (const ObjectRef &other) const
	Comparator. More...
bool	operator< (const ObjectRef &other) const
	Comparator. More...
bool	defined () const
const Object *	get () const
const Object *	operator-> () const
bool	unique () const
int	use_count () const
template<typename ObjectType >
const ObjectType *	as () const
	Try to downcast the internal Object to a raw pointer of a corresponding type. More...

Additional Inherited Members
Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool	_type_is_nullable = true
Protected Member Functions inherited from tvm::runtime::ObjectRef
Object *	get_mutable () const
Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T	DowncastNoCheck (ObjectRef ref)
	Internal helper function downcast a ref without check. More...
static void	FFIClearAfterMove (ObjectRef *ref)
	Clear the object ref data field without DecRef after we successfully moved the field. More...
template<typename ObjectType >
static ObjectPtr< ObjectType >	GetDataPtr (const ObjectRef &ref)
	Internal helper function get data_ as ObjectPtr of ObjectType. More...
Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Object >	data_
	Internal pointer that backs the reference. 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