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

Public Member Functions
	Operation (ffi::UnsafeInit tag)
const OperationNode *	operator-> () const
	access the internal node container
Tensor	output (size_t i) const
	get the i-th output of the operation.

Static Public Attributes
static constexpr bool	_type_is_nullable = false

Detailed Description

Operation that produces tensors.

Member Typedef Documentation

ContainerType

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

specify container node

Constructor & Destructor Documentation

Operation()

tvm::te::Operation::Operation ( ffi::UnsafeInit  tag )

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.

Member Data Documentation

_type_is_nullable

constexpr bool tvm::te::Operation::_type_is_nullable = false

staticconstexpr

---

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

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